diff --git a/src/api/checkpoint/index.test.js b/src/api/checkpoint/index.test.js
index 4dfb18c..2e1b62f 100644
--- a/src/api/checkpoint/index.test.js
+++ b/src/api/checkpoint/index.test.js
@@ -9,7 +9,7 @@ const ioMock = {
   emit: jest.fn()
 }
 
-socket.setUpSocketRoom(ioMock)
+socket.register(ioMock)
 
 const app = () => express(routes)
 
diff --git a/src/api/checkpoint/model.test.js b/src/api/checkpoint/model.test.js
index 31e2f65..4a706fd 100644
--- a/src/api/checkpoint/model.test.js
+++ b/src/api/checkpoint/model.test.js
@@ -7,7 +7,7 @@ const ioMock = {
   emit: jest.fn()
 }
 
-socket.setUpSocketRoom(ioMock)
+socket.register(ioMock)
 
 let checkpoint
 
diff --git a/src/api/journey/model.test.js b/src/api/journey/model.test.js
index 2bbe394..167d915 100644
--- a/src/api/journey/model.test.js
+++ b/src/api/journey/model.test.js
@@ -7,7 +7,7 @@ const ioMock = {
   emit: jest.fn()
 }
 
-socket.setUpSocketRoom(ioMock)
+socket.register(ioMock)
 
 let user, journey, checkpoint1, checkpoint2