From 2d06cdfc1c027b1454724ad29a9bc05be8d5801a Mon Sep 17 00:00:00 2001
From: Mark Erikson <mark@isquaredsoftware.com>
Date: Mon, 5 Nov 2018 23:52:53 -0500
Subject: [PATCH] Fix custom context as a prop usage

---
 src/components/connectAdvanced.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/components/connectAdvanced.js b/src/components/connectAdvanced.js
index 007eed098..f7bec9409 100644
--- a/src/components/connectAdvanced.js
+++ b/src/components/connectAdvanced.js
@@ -110,7 +110,6 @@ export default function connectAdvanced(
 
     if(pure) {
       OuterBaseComponent = PureComponent
-      //FinalWrappedComponent = React.memo(WrappedComponent)
     }
 
     class PureWrapper extends Component {
@@ -207,10 +206,12 @@ export default function connectAdvanced(
       }
 
       render() {
+        const ContextToUse = this.props.context || Context
+
         return (
-          <Context.Consumer>
+          <ContextToUse.Consumer>
             {this.renderWrappedComponent}
-          </Context.Consumer>
+          </ContextToUse.Consumer>
         )
       }
     }