diff --git a/scaffold/src/modules/RemoveMe/graphql/queries/HelloQuery.spec.ts b/scaffold/src/modules/RemoveMe/graphql/queries/HelloQuery.spec.ts
index f44a9051..e06ba5f7 100644
--- a/scaffold/src/modules/RemoveMe/graphql/queries/HelloQuery.spec.ts
+++ b/scaffold/src/modules/RemoveMe/graphql/queries/HelloQuery.spec.ts
@@ -1,9 +1,9 @@
 import td from "testdouble";
-import { GqlContext, QueryHelloArgs } from "~generated/graphql/types";
-import { HelloQuery } from "./HelloQuery";
-
-const testHello = (variables: QueryHelloArgs, context: GqlContext) =>
-  HelloQuery({}, variables, context, null);
+import {
+  GqlContext,
+  QueryHelloArgs,
+  testHello,
+} from "~generated/graphql/helpers/HelloQuerySpecWrapper";
 
 test("Hello", async () => {
   const context = td.object<GqlContext>();
diff --git a/test-module/graphql/scalars/Odd.ts b/test-module/graphql/scalars/Odd.ts
index 0582e26a..83365aba 100644
--- a/test-module/graphql/scalars/Odd.ts
+++ b/test-module/graphql/scalars/Odd.ts
@@ -1,6 +1,9 @@
 import { GraphQLScalarType, Kind } from "graphql";
 
-function oddValue(value) {
+function oddValue(value: unknown) {
+  if (typeof value !== "number") {
+    return null;
+  }
   return value % 2 === 1 ? value : null;
 }