1 files changed, 4 insertions, 2 deletions
diff --git a/src/systems/keyboardInput.js b/src/systems/keyboardInput.js
index 60a62a1..5ce6737 100644
--- a/src/systems/keyboardInput.js
+++ b/src/systems/keyboardInput.js
@@ -1,4 +1,4 @@
-game.system.KeyboardInput = () => {
+game.system.KeyboardInput = (undoSystem) => {
   "use strict";
   const keys = {};
   const keyPress = (event) => {
@@ -20,9 +20,11 @@ game.system.KeyboardInput = () => {
         } else if (controls.includes('down') && keys['ArrowDown']) {
           entity.addComponent(game.components.Momentum({ dx: 0, dy: 1 }));
         }
-
       }
     }
+    if (keys['z']) {
+      undoSystem.undo(entities);
+    }
     Object.keys(keys).map((key) => delete keys[key]);
 
     return new Set();