diff --git a/tests/test_engine.py b/tests/test_engine.py
index 416fe713750291f5c8332d1dc6a01b19e8497b8b..5e88a5c055afab374552f6e0f3ff258427055581 100644
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@@ -123,14 +123,15 @@ def test_valid_error() -> None:
     inp.marginals[0].dist_type = 'normal'
     inp.marginals[0].parameters = [0, 1]
     mm = PCE(inp)
-    mm.fit([[0.0, 0.0], [1.0, 1.0]], {'Z': [[0.5], [0.5]]})  # Shape (2, 2)
+    mm.fit([[0.0], [1.0]], {'Z': [[0.5], [0.5]]})  # Shape (2, 2)
 
     expdes = ExpDesigns(inp)
     mod = PL()
 
+    
     # Update valid_samples to match input size during fitting
-    expdes.valid_samples = np.array([[0.5, 0.5]])  # Shape (1, 2)
-    expdes.valid_model_runs = {'Z': np.array([[0.5], [0.5]])}
+    expdes.valid_samples = np.array([[0.5]])  # Shape (1, 2)
+    expdes.valid_model_runs = {'Z': np.array([[0.5]])}
     mod.mc_reference['mean'] = [0.5]
     mod.mc_reference['std'] = [0.0]
     mod.output.names = ['Z']