diff --git a/src/eynollah/model_zoo/model_zoo.py b/src/eynollah/model_zoo/model_zoo.py
index 2bac7f3..d5e69a2 100644
--- a/src/eynollah/model_zoo/model_zoo.py
+++ b/src/eynollah/model_zoo/model_zoo.py
@@ -207,13 +207,14 @@ class EynollahModelZoo:
         model_path = self.model_path(model_category, model_variant)
         try:
             if model_path.is_dir() and not (model_path / "keras_metadata.pb").exists():
+                # short-cut to avoid warning for exported models
                 raise ValueError()
             model = load_model(model_path, compile=False)
             model.make_predict_function()
         except (AttributeError, ValueError):
             model = tf.saved_model.load(model_path)
             model.predict_on_batch = model.serve
-            model.input_shape = model.signatures.get('serving_default').inputs[0].shape
+            model.input_shape = tuple(model.signatures.get('serving_default').inputs[0].shape)
         model._name = model_category
         if resized:
             model = wrap_layout_model_resized(model)