From a9384b03cbd316c37a156536fac6a6f206260d2e Mon Sep 17 00:00:00 2001
From: rakannimer <rakannimer@gmail.com>
Date: Sun, 5 Apr 2020 11:28:46 +0300
Subject: [PATCH] fix(gatsby-theme-docz): fix theme components customization
 bug

---
 core/gatsby-theme-docz/src/base/Layout.js | 9 ++++++---
 core/gatsby-theme-docz/src/index.js       | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/core/gatsby-theme-docz/src/base/Layout.js b/core/gatsby-theme-docz/src/base/Layout.js
index 59c52605a..1ec8553f4 100644
--- a/core/gatsby-theme-docz/src/base/Layout.js
+++ b/core/gatsby-theme-docz/src/base/Layout.js
@@ -2,6 +2,7 @@ import React, { Fragment } from 'react'
 import PropTypes from 'prop-types'
 import { useComponents } from 'docz'
 import { propEq, get } from 'lodash/fp'
+import { MDXProvider } from '@mdx-js/react'
 
 import { useDbQuery } from '../hooks/useDbQuery'
 import Wrapper from '../wrapper'
@@ -17,9 +18,11 @@ const Route = ({ children, entry, isTransclusion, ...defaultProps }) => {
   return isTransclusion ? (
     children
   ) : (
-    <Wrapper>
-      <Layout {...props}>{children}</Layout>
-    </Wrapper>
+    <MDXProvider components={components}>
+      <Wrapper>
+        <Layout {...props}>{children}</Layout>
+      </Wrapper>
+    </MDXProvider>
   )
 }
 
diff --git a/core/gatsby-theme-docz/src/index.js b/core/gatsby-theme-docz/src/index.js
index 7eca87699..fe69ce6b0 100644
--- a/core/gatsby-theme-docz/src/index.js
+++ b/core/gatsby-theme-docz/src/index.js
@@ -9,7 +9,7 @@ import components from '~components'
 const Theme = ({ children }) => {
   const config = useConfig()
   return (
-    <ThemeProvider theme={config.themeConfig} components={components}>
+    <ThemeProvider theme={config.themeConfig}>
       <ComponentsProvider components={components}>
         <Styled.root>{children}</Styled.root>
       </ComponentsProvider>