1 files changed, 1 insertions, 8 deletions
diff --git a/lib/withData.js b/lib/withData.js
index 6b37792..ab9fe78 100644
--- a/lib/withData.js
+++ b/lib/withData.js
@@ -49,14 +49,7 @@ export default ComposedComponent =>
         Head.rewind();
 
         // Extract query data from the Apollo store
-        const state = apollo.getInitialState();
-
-        serverState = {
-          apollo: {
-            // Only include the Apollo data state
-            data: state.data,
-          },
-        };
+        serverState = apollo.cache.extract();
       }
 
       return {