diff --git a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
index 265ea5a79..eadacd1e2 100755
--- a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
+++ b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
@@ -34,7 +34,7 @@ def __init__(self):
 
     def internal_init(self, application, options = {}):
         layers = [
-            tensorflow.keras.layers.Input(int(options['input_shape'])),
+            tensorflow.keras.layers.Input((int(options['input_shape']), )),
             tensorflow.keras.layers.Dense(int(options['output_shape']), activation = 'softmax'),
         ]