diff --git a/test/python/transpiler/test_remove_final_reset.py b/test/python/transpiler/test_remove_final_reset.py index 371ff3f8405f..22ac234a81f5 100644 --- a/test/python/transpiler/test_remove_final_reset.py +++ b/test/python/transpiler/test_remove_final_reset.py @@ -15,6 +15,7 @@ import unittest from qiskit import QuantumRegister, QuantumCircuit +from qiskit.passmanager.flow_controllers import DoWhileController from qiskit.transpiler import PassManager from qiskit.transpiler.passes import RemoveFinalReset, DAGFixedPoint from qiskit.converters import circuit_to_dag @@ -102,8 +103,10 @@ def test_two_resets(self): pass_manager = PassManager() pass_manager.append( - [RemoveFinalReset(), DAGFixedPoint()], - do_while=lambda property_set: not property_set["dag_fixed_point"], + DoWhileController( + [RemoveFinalReset(), DAGFixedPoint()], + do_while=lambda property_set: not property_set["dag_fixed_point"], + ) ) after = pass_manager.run(circuit)