diff --git a/property_tests/tests/post_upgrade_method/test/test.ts b/property_tests/tests/post_upgrade_method/test/test.ts
index febaf575ac..13e1913c49 100644
--- a/property_tests/tests/post_upgrade_method/test/test.ts
+++ b/property_tests/tests/post_upgrade_method/test/test.ts
@@ -121,7 +121,7 @@ function generateGetPostDeployValuesCanisterMethod(
     globalPostDeployVariableNames: string[]
 ): QueryMethod {
     return {
-        imports: new Set(['Tuple', 'bool']),
+        imports: new Set(['Tuple', 'bool', 'query']),
         globalDeclarations: [],
         sourceCode: /*TS*/ `getPostDeployValues: query(
             [],