fix: tests

intezer · Feb 22, 2023 · 8d2fd6a · 8d2fd6a
1 parent 07022dc
commit 8d2fd6a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/intezer_sdk/_operation.py b/intezer_sdk/_operation.py
@@ -3,7 +3,7 @@
 from typing import Optional
 from typing import Union
 
-from intezer_sdk.api import IntezerApi
+from intezer_sdk._api import IntezerApi
 from intezer_sdk.operation import Operation
 
 
@@ -14,7 +14,7 @@ def handle_operation(operations: Dict[str, Operation],
                      wait: Union[bool, int],
                      wait_timeout: Optional[datetime.timedelta]) -> Operation:
     if operation not in operations:
-        operations[operation] = Operation(result_url, operation, api=api)
+        operations[operation] = Operation(result_url, operation, api=api.api)
 
         if wait:
             if isinstance(wait, bool):