fix import location

demergent-labs · Aug 20, 2024 · 550ecf0 · 550ecf0
1 parent 4b7eb36
commit 550ecf0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/property_tests/arbitraries/canister_arb.ts b/property_tests/arbitraries/canister_arb.ts
@@ -142,6 +142,7 @@ function generateSourceCode(
     ]
         .sort((a, b) => a.localeCompare(b, 'en', { sensitivity: 'base' }))
         .join();
+    const importLocation = `azle${api === 'functional' ? '/experimental' : ''}`;
 
     const declarationsFromCanisterMethods = canisterMethods.flatMap(
         (method) => method.globalDeclarations
@@ -179,7 +180,7 @@ function generateSourceCode(
     `;
 
     return /*TS*/ `
-        import { ${imports} } from 'azle';
+        import { ${imports} } from '${importLocation}';
 
         // @ts-ignore
         import deepEqual from 'deep-is';