diff --git a/qiskit_ibm_transpiler/wrappers/ai_local_linear_function_synthesis.py b/qiskit_ibm_transpiler/wrappers/ai_local_linear_function_synthesis.py
index 0c5a637..c2e2f98 100644
--- a/qiskit_ibm_transpiler/wrappers/ai_local_linear_function_synthesis.py
+++ b/qiskit_ibm_transpiler/wrappers/ai_local_linear_function_synthesis.py
@@ -14,7 +14,7 @@
 import networkx as nx
 from networkx.exception import NetworkXError
 
-from ai_transpiler import LinearFunctionInference
+from qiskit_ibm_ai_local_transpiler import AILinearFunctionInference
 
 from typing import Union, List
 
@@ -112,7 +112,7 @@ def get_synthesized_linear_function_circuits(
         # Generate the Clifford from the dictionary to send it to the model and permute it
         clifford = perm_cliff(Clifford.from_dict(clifford_dicts[index]), subgraph_perm)
 
-        synthesized_linear_function = LinearFunctionInference().synthesize(
+        synthesized_linear_function = AILinearFunctionInference().synthesize(
             cliff=clifford, coupling_map_hash=cmap_hash
         )