diff --git a/tests/test_quimb_backend.py b/tests/test_quimb_backend.py
index e01cde47..81a0e2b7 100644
--- a/tests/test_quimb_backend.py
+++ b/tests/test_quimb_backend.py
@@ -53,7 +53,7 @@ def test_eval(nqubits: int, tolerance: float, is_mps: bool):
     # Test quimb
     result_tn = qibotn.eval_qu.dense_vector_tn_qu(
             qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend
-        )
+        ).flatten()
    
 
     assert np.allclose(result_sv, result_tn,