diff --git a/docs/src/App.js b/docs/src/App.js
index 98e4835cdd..8aeda90e42 100644
--- a/docs/src/App.js
+++ b/docs/src/App.js
@@ -1,5 +1,5 @@
import React from 'react'
-import { hot } from 'react-hot-loader'
+import { hot } from 'react-hot-loader/root'
import { Router, Switch } from 'react-static'
import Routes from 'react-static-routes'
@@ -11,4 +11,4 @@ const App = () => (
)
-export default hot(module)(App)
+export default hot(App)
diff --git a/docs/src/components/Document.js b/docs/src/components/Document.js
index 36a9692eba..b60749d83c 100644
--- a/docs/src/components/Document.js
+++ b/docs/src/components/Document.js
@@ -16,9 +16,7 @@ const Document = ({ Body, children, Head, Html, siteData: { dev, versions } }) =
href={`https://cdn.jsdelivr.net/npm/semantic-ui@${versions.sui}/dist/semantic.min.css`}
/>
-
+
@@ -57,9 +55,15 @@ const Document = ({ Body, children, Head, Html, siteData: { dev, versions } }) =
}.js`}
/>