diff --git a/dist/media-query-provider.js b/dist/media-query-provider.js
index d4add21..c3ab5c0 100644
--- a/dist/media-query-provider.js
+++ b/dist/media-query-provider.js
@@ -105,6 +105,7 @@ var MediaQueryProvider = function (_React$Component) {
 
       // need to rerender with correct media if it didnt match up with initial
       if (!(0, _shallowequal2.default)(media, this.state.media)) {
+        this.currentMediaState = media;
         this.setState({ media: media }); // eslint-disable-line react/no-did-mount-set-state
       }
     }
diff --git a/src/media-query-provider.js b/src/media-query-provider.js
index ac31413..c0624a3 100644
--- a/src/media-query-provider.js
+++ b/src/media-query-provider.js
@@ -66,6 +66,7 @@ class MediaQueryProvider extends React.Component {
 
     // need to rerender with correct media if it didnt match up with initial
     if (!shallowequal(media, this.state.media)) {
+      this.currentMediaState = media;
       this.setState({ media }); // eslint-disable-line react/no-did-mount-set-state
     }
   }