diff --git a/src/pages/404.tsx b/src/pages/404.tsx
index a8ee0ee798d6d95f21d445e89eba07ae79032d32..80fe09c9427ee294a7c8385cf8b1083b0dac3e4d 100644
--- a/src/pages/404.tsx
+++ b/src/pages/404.tsx
@@ -1,5 +1,6 @@
import React from 'react';
import { Button } from 'antd';
+import router from 'umi/router';
// 这里应该使用 antd 的 404 result 组件,
// 但是还没发布,先来个简单的。
@@ -17,7 +18,9 @@ const NoFoundPage: React.FC<{}> = () => (
Sorry, the page you visited does not exist.
- + );