diff --git a/src/resolvers/__tests__/EtaResolverTests.test.ts b/src/resolvers/__tests__/EtaResolverTests.test.ts
index e6bbe20..f584b8c 100644
--- a/src/resolvers/__tests__/EtaResolverTests.test.ts
+++ b/src/resolvers/__tests__/EtaResolverTests.test.ts
@@ -6,7 +6,8 @@ import {
   addMockShuttleToRepository,
   addMockStopToRepository,
 } from "../../../testHelpers/repositorySetupHelpers";
-import assert = require("node:assert");
+import { ExternalSourceETARepository } from "../../repositories/shuttle/eta/ExternalSourceETARepository";
+import assert from "node:assert";
 
 describe("EtaResolvers", () => {
   const holder = setupTestServerHolder();
@@ -19,7 +20,7 @@ describe("EtaResolvers", () => {
   beforeEach(async () => {
     mockShuttle = await addMockShuttleToRepository(context.systems[0].shuttleRepository, context.systems[0].id);
     mockStop = await addMockStopToRepository(context.systems[0].shuttleRepository, context.systems[0].id);
-    expectedEta = await addMockEtaToRepository(context.systems[0].shuttleRepository, mockStop.id, mockShuttle.id);
+    expectedEta = await addMockEtaToRepository(context.systems[0].etaRepository as ExternalSourceETARepository, mockStop.id, mockShuttle.id);
   });
 
   async function getResponseForEtaQuery(query: string) {
diff --git a/testHelpers/repositorySetupHelpers.ts b/testHelpers/repositorySetupHelpers.ts
index 7be3d1c..657fd4d 100644
--- a/testHelpers/repositorySetupHelpers.ts
+++ b/testHelpers/repositorySetupHelpers.ts
@@ -5,6 +5,7 @@ import {
   generateMockStops,
 } from "./mockDataGenerators";
 import { ShuttleGetterSetterRepository } from "../src/repositories/shuttle/ShuttleGetterSetterRepository";
+import { ExternalSourceETARepository } from "../src/repositories/shuttle/eta/ExternalSourceETARepository";
 
 export async function addMockRouteToRepository(repository: ShuttleGetterSetterRepository, systemId: string) {
   const mockRoutes = generateMockRoutes();
@@ -32,7 +33,7 @@ export async function addMockShuttleToRepository(repository: ShuttleGetterSetter
   return mockShuttle;
 }
 
-export async function addMockEtaToRepository(repository: ShuttleGetterSetterRepository, stopId: string, shuttleId: string) {
+export async function addMockEtaToRepository(repository: ExternalSourceETARepository, stopId: string, shuttleId: string) {
   const etas = generateMockEtas();
   const expectedEta = etas[0];
   expectedEta.stopId = stopId;