diff --git a/spec/e2e/game_spec.js b/spec/e2e/game_spec.js
index 23420ea..ce43646 100644
--- a/spec/e2e/game_spec.js
+++ b/spec/e2e/game_spec.js
@@ -28,6 +28,9 @@ describe('game page', () => {
                 once: function (message, handler) {
                     this.eventHandlers[message] = handler;
                 },
+                timeout: (duration) => {
+                    return mockSocket;
+                },
                 emit: function (eventName, ...args) {
                     switch (args[0]) { // eventName is currently always "inGameMessage" - the first arg after that is the specific message type
                         case globals.EVENT_IDS.FETCH_GAME_STATE:
@@ -93,6 +96,9 @@ describe('game page', () => {
                 on: function (message, handler) {
                     this.eventHandlers[message] = handler;
                 },
+                timeout: (duration) => {
+                    return mockSocket;
+                },
                 emit: function (eventName, ...args) {
                     switch (args[0]) { // eventName is currently always "inGameMessage" - the first arg after that is the specific message type
                         case globals.EVENT_IDS.FETCH_GAME_STATE:
@@ -167,6 +173,9 @@ describe('game page', () => {
                 on: function (message, handler) {
                     this.eventHandlers[message] = handler;
                 },
+                timeout: (duration) => {
+                    return mockSocket;
+                },
                 once: function (message, handler) {
                     this.eventHandlers[message] = handler;
                 },