diff --git a/test/resolvers/SystemResolverTests.test.ts b/test/resolvers/SystemResolverTests.test.ts
index af37384..bf86d8e 100644
--- a/test/resolvers/SystemResolverTests.test.ts
+++ b/test/resolvers/SystemResolverTests.test.ts
@@ -27,9 +27,7 @@ describe("SystemResolvers", () => {
         systemId: mockSystem.id,
       },
     }, {
-      contextValue: {
-        shuttleRepository: context.systems[0].shuttleRepository,
-      },
+      contextValue: context,
     });
   }
 
@@ -109,9 +107,7 @@ describe("SystemResolvers", () => {
           stopId: stopId,
         },
       }, {
-        contextValue: {
-          shuttleRepository: context.systems[0].shuttleRepository,
-        }
+        contextValue: context,
       });
     }