diff --git a/stories/useStateWithHistory.story.tsx b/stories/useStateWithHistory.story.tsx
index 0549350816..1ce477f8d1 100644
--- a/stories/useStateWithHistory.story.tsx
+++ b/stories/useStateWithHistory.story.tsx
@@ -24,7 +24,7 @@ const Demo = () => {
         return;
       }
 
-      window.history.back(stepSize);
+      history.back(stepSize);
     },
     [history, stepSize]
   );
@@ -35,7 +35,7 @@ const Demo = () => {
         return;
       }
 
-      window.history.forward(stepSize);
+      history.forward(stepSize);
     },
     [history, stepSize]
   );
@@ -60,12 +60,12 @@ const Demo = () => {
         Current state: <span>{state}</span>
       </div>
       <div style={{ marginTop: 8 }}>
-        <button onClick={handleBackClick} disabled={!window.history.position}>
+        <button onClick={handleBackClick} disabled={!history.position}>
           &lt; Back
         </button>
         <button
           onClick={handleForwardClick}
-          disabled={window.history.position >= window.history.window.history.length - 1}>
+          disabled={history.position >= history.history.length - 1}>
           Forward &gt;
         </button>
         &nbsp; Step size:&nbsp;
@@ -76,7 +76,7 @@ const Demo = () => {
         <div>Current history</div>
         <div
           dangerouslySetInnerHTML={{
-            __html: JSON.stringify(window.history.history, null, 2)
+            __html: JSON.stringify(history.history, null, 2)
               .replace(/\n/g, '<br/>')
               .replace(/ /g, '&nbsp;'),
           }}