diff --git a/tests/transformation/test_absorb_transp_into_flatten.py b/tests/transformation/test_absorb_transp_into_flatten.py
index 609f28139f30a19fe54f857c2fa14018a4b2211d..fbfa15277717c554da01e38608601997407803b2 100644
--- a/tests/transformation/test_absorb_transp_into_flatten.py
+++ b/tests/transformation/test_absorb_transp_into_flatten.py
@@ -65,7 +65,7 @@ def test_absorb_transp_into_flatten(perm, shape, ishape, data_layout):
     inp_values = np.random.uniform(low=-1, high=1, size=tuple(ishape)).astype(
         np.float32
     )
-    idict = {"inp": inp_values}
+    idict = {model.graph.input[0].name: inp_values}
     assert oxe.compare_execution(model, model_transformed, idict)
 
     # only some of the parameter combinations lead to a graph that will be changed when