diff --git a/src/demo/ui.jsx b/src/demo/ui.jsx
index afd23cae..38985fa9 100644
--- a/src/demo/ui.jsx
+++ b/src/demo/ui.jsx
@@ -935,7 +935,7 @@ class Pond extends React.Component {
                 <Button
                   style={styles.playAgainButton}
                   onClick={() => {
-                    resetTraining();
+                    resetTraining(state);
                     toMode(Modes.Words);
                   }}
                 >