diff --git a/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs b/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs
index c3ad92dd..53a13394 100644
--- a/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs
+++ b/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs
@@ -135,7 +135,7 @@ namespace TensorFlowNET.Keras.UnitTest
         {
             var ed = keras.layers.EinsumDense(
                 equation: "...b,bc->...c",
-                output_shape: (-1, 4),
+                output_shape: 4,
                 bias_axes: "c",
                 bias_initializer: tf.constant_initializer(0.03),
                 kernel_initializer: tf.constant_initializer(0.5)