diff --git a/dist/media-query-provider.js b/dist/media-query-provider.js
index acbe4b8..43912f0 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 0fe7818..b64e4ce 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
     }
   }