From 550ecf0435c16d419279b0e5b974167d8c853e33 Mon Sep 17 00:00:00 2001
From: Benjamin DeMann <benjamin.demann@gmail.com>
Date: Tue, 20 Aug 2024 13:45:01 -0600
Subject: [PATCH] fix import location

---
 property_tests/arbitraries/canister_arb.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/property_tests/arbitraries/canister_arb.ts b/property_tests/arbitraries/canister_arb.ts
index 605a3aa309..4d642ad680 100644
--- 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';