diff --git a/property_tests/tests/canister_methods/pre_upgrade/test/test.ts b/property_tests/tests/canister_methods/pre_upgrade/test/test.ts
index 6a2c8912d7..38506f73d6 100644
--- a/property_tests/tests/canister_methods/pre_upgrade/test/test.ts
+++ b/property_tests/tests/canister_methods/pre_upgrade/test/test.ts
@@ -83,7 +83,7 @@ runPropTests(CanisterArb(CanisterConfigArb));
 
 function generateGetPreUpgradeExecutedCanisterMethod(): QueryMethod {
     return {
-        imports: new Set(['bool', 'query', 'StableBTreeMap']),
+        imports: new Set(['bool', 'query', 'StableBTreeMap', 'text']),
         globalDeclarations: [],
         sourceCode: /*TS*/ `getPreUpgradeExecuted: query([], bool,() => {
             return stable.get(PRE_UPGRADE_HOOK_EXECUTED).Some === true