diff --git a/tests/mlmodel/test_random_forest.py b/tests/mlmodel/test_random_forest.py
index baee9d90..227fc528 100644
--- a/tests/mlmodel/test_random_forest.py
+++ b/tests/mlmodel/test_random_forest.py
@@ -151,7 +151,6 @@ def test_fit_class_random_forest_from_graph(
                                 }
                             }
                         },
-                        "target_dimension": "bands",
                     },
                 },
                 "fitclassrandomforest1": {