diff --git a/test/testHelpers/repositorySetupHelpers.ts b/test/testHelpers/repositorySetupHelpers.ts
index d9e48e3..4c95363 100644
--- a/test/testHelpers/repositorySetupHelpers.ts
+++ b/test/testHelpers/repositorySetupHelpers.ts
@@ -1,4 +1,3 @@
-import { UnoptimizedInMemoryRepository } from "../../src/repositories/UnoptimizedInMemoryRepository";
 import {
   generateMockEtas,
   generateMockRoutes,
@@ -6,8 +5,9 @@ import {
   generateMockStops,
   generateMockSystems
 } from "./mockDataGenerators";
+import { GetterSetterRepository } from "../../src/repositories/GetterSetterRepository";
 
-export async function addMockSystemToRepository(repository: UnoptimizedInMemoryRepository) {
+export async function addMockSystemToRepository(repository: GetterSetterRepository) {
   const mockSystems = generateMockSystems();
   const mockSystem = mockSystems[0];
   mockSystem.id = "1";
@@ -16,7 +16,7 @@ export async function addMockSystemToRepository(repository: UnoptimizedInMemoryR
   return mockSystem;
 }
 
-export async function addMockRouteToRepository(repository: UnoptimizedInMemoryRepository, systemId: string) {
+export async function addMockRouteToRepository(repository: GetterSetterRepository, systemId: string) {
   const mockRoutes = generateMockRoutes();
   const mockRoute = mockRoutes[0];
   mockRoute.systemId = systemId;
@@ -25,7 +25,7 @@ export async function addMockRouteToRepository(repository: UnoptimizedInMemoryRe
   return mockRoute;
 }
 
-export async function addMockStopToRepository(repository: UnoptimizedInMemoryRepository, systemId: string) {
+export async function addMockStopToRepository(repository: GetterSetterRepository, systemId: string) {
   const mockStops = generateMockStops();
   const mockStop = mockStops[0];
   mockStop.systemId = systemId;
@@ -34,7 +34,7 @@ export async function addMockStopToRepository(repository: UnoptimizedInMemoryRep
   return mockStop;
 }
 
-export async function addMockShuttleToRepository(repository: UnoptimizedInMemoryRepository, systemId: string) {
+export async function addMockShuttleToRepository(repository: GetterSetterRepository, systemId: string) {
   const mockShuttles = generateMockShuttles();
   const mockShuttle = mockShuttles[0];
   mockShuttle.systemId = systemId;
@@ -42,7 +42,7 @@ export async function addMockShuttleToRepository(repository: UnoptimizedInMemory
   return mockShuttle;
 }
 
-export async function addMockEtaToRepository(repository: UnoptimizedInMemoryRepository, stopId: string, shuttleId: string) {
+export async function addMockEtaToRepository(repository: GetterSetterRepository, stopId: string, shuttleId: string) {
   const etas = generateMockEtas();
   const expectedEta = etas[0];
   expectedEta.stopId = stopId;