diff --git a/packages/antdsite/index.js b/packages/antdsite/index.js
index 2d16fa1..405b225 100644
--- a/packages/antdsite/index.js
+++ b/packages/antdsite/index.js
@@ -1 +1 @@
-export { PageContext } from './templates/PageContext';
+export { PageContext } from './src/templates/PageContext';
diff --git a/packages/antdsite/src/default-theme/components/content/Article.tsx b/packages/antdsite/src/default-theme/components/content/Article.tsx
index a523e92..76c998b 100644
--- a/packages/antdsite/src/default-theme/components/content/Article.tsx
+++ b/packages/antdsite/src/default-theme/components/content/Article.tsx
@@ -4,7 +4,7 @@ import EditButton from './EditButton';
 import { OneToc } from '../../../templates';
 import moment from 'moment';
 import AvatarList from './AvatarList';
-import { PageContext } from 'antdsite/src/templates/PageContext';
+import { PageContext } from 'antdsite';
 import SEO from '../SEO/SEO';
 import MDXRenderer from 'gatsby-mdx-fix/mdx-renderer';
 
diff --git a/packages/antdsite/src/default-theme/layout/header.tsx b/packages/antdsite/src/default-theme/layout/header.tsx
index cfbda12..6c57a63 100644
--- a/packages/antdsite/src/default-theme/layout/header.tsx
+++ b/packages/antdsite/src/default-theme/layout/header.tsx
@@ -3,7 +3,7 @@ import React from 'react';
 import { Link } from 'gatsby';
 import * as utils from '../components/utils';
 import { Row, Col, Icon, Input, Menu, Button, Popover, Dropdown, Affix, Badge } from 'antd';
-import { PageContext } from 'antdsite/src/templates/PageContext';
+import { PageContext } from 'antdsite';
 import SearchBox from '../components/search-box';
 
 interface HeaderProps {
diff --git a/packages/antdsite/src/default-theme/layout/home.tsx b/packages/antdsite/src/default-theme/layout/home.tsx
index 04a3e8c..c56594c 100644
--- a/packages/antdsite/src/default-theme/layout/home.tsx
+++ b/packages/antdsite/src/default-theme/layout/home.tsx
@@ -4,12 +4,12 @@ import SEO from '../components/SEO/SEO';
 import Banner from '../components/home/Banner.jsx';
 import Features from '../components/home/Features.jsx';
 import HomeRest from '../components/home/HomeRest.jsx';
-import { PageContext } from 'antdsite/src/templates/PageContext';
+import { PageContext } from 'antdsite';
 
 function Home(props: any) {
   return (
     <PageContext.Consumer>
-      {value => {
+      {(value: any) => {
         const { currentLocaleWebConfig } = value;
         const { title, description } = currentLocaleWebConfig as any;
 
diff --git a/packages/antdsite/src/default-theme/layout/main-content.tsx b/packages/antdsite/src/default-theme/layout/main-content.tsx
index 6dd7508..2f969fc 100644
--- a/packages/antdsite/src/default-theme/layout/main-content.tsx
+++ b/packages/antdsite/src/default-theme/layout/main-content.tsx
@@ -5,7 +5,7 @@ import classNames from 'classnames';
 import MobileMenu from 'rc-drawer';
 import Article from '../components/content/Article';
 import { PageInfo, getPageTitle } from '../components/utils';
-import { PageContext } from 'antdsite/src/templates/PageContext';
+import { PageContext } from 'antdsite';
 
 const { SubMenu } = Menu;
 
diff --git a/packages/antdsite/src/module.d.ts b/packages/antdsite/src/module.d.ts
index 0bc50de..70963e7 100644
--- a/packages/antdsite/src/module.d.ts
+++ b/packages/antdsite/src/module.d.ts
@@ -1,3 +1,4 @@
 declare module '*.less';
 declare module 'gatsby-mdx-fix/mdx-renderer';
+declare module 'antdsite';
 declare module '*.jsx';