diff --git a/src/common/components/layout/CustomHeader.tsx b/src/common/components/layout/CustomHeader.tsx
index f3106ae..247ebb1 100644
--- a/src/common/components/layout/CustomHeader.tsx
+++ b/src/common/components/layout/CustomHeader.tsx
@@ -1,7 +1,7 @@
 import React from 'react';
 import { Layout, Dropdown } from 'antd';
 import type { MenuProps } from 'antd';
-import { history } from '@umijs/max';
+import { useLocation } from '@umijs/max';
 import AvatarProps from './AvatarProps';
 import { useMyState } from '@/common';
 
@@ -14,7 +14,7 @@ interface CustomHeaderProps {
 
 export const CustomHeader: React.FC<CustomHeaderProps> = ({ collapsed, toggle }) => {
   const { snap } = useMyState();
-  const { pathname } = history.useLocation();
+  const { pathname } = useLocation();
 
   // 在登录页面不显示header
   if (pathname === '/login') {