diff --git a/src/openfermion/circuits/primitives/state_preparation_test.py b/src/openfermion/circuits/primitives/state_preparation_test.py
index e9bc83a42..9c1375281 100644
--- a/src/openfermion/circuits/primitives/state_preparation_test.py
+++ b/src/openfermion/circuits/primitives/state_preparation_test.py
@@ -57,7 +57,7 @@ def test_prepare_gaussian_state(
     )
     if isinstance(initial_state, list):
         initial_state = sum(1 << (n_qubits - 1 - i) for i in initial_state)
-    state = circuit.final_state_vector(initial_state=initial_state)
+    state = circuit.final_state_vector(initial_state=initial_state, qubit_order=qubits)
 
     # Check that the result is an eigenstate with the correct eigenvalue
     numpy.testing.assert_allclose(quad_ham_sparse.dot(state), energy * state, atol=atol)
@@ -104,7 +104,7 @@ def test_prepare_gaussian_state_with_spin_symmetry(
 
     if isinstance(initial_state, list):
         initial_state = sum(1 << (n_qubits - 1 - i) for i in initial_state)
-    state = circuit.final_state_vector(initial_state=initial_state)
+    state = circuit.final_state_vector(initial_state=initial_state, qubit_order=qubits)
 
     # Check that the result is an eigenstate with the correct eigenvalue
     numpy.testing.assert_allclose(quad_ham_sparse.dot(state), energy * state, atol=atol)
@@ -150,6 +150,6 @@ def test_prepare_slater_determinant(
     )
     if isinstance(initial_state, list):
         initial_state = sum(1 << (n_qubits - 1 - i) for i in initial_state)
-    state = circuit.final_state_vector(initial_state=initial_state)
+    state = circuit.final_state_vector(initial_state=initial_state, qubit_order=qubits)
 
     assert cirq.allclose_up_to_global_phase(state, correct_state, atol=atol)