diff --git a/tests/test_modelwrapper.py b/tests/test_modelwrapper.py
index 3dfeee89efd82f67f4c54e75fdac56959dd18276..e3104ffcf13728d1e970e9d40371a5cc376d7cbb 100644
--- a/tests/test_modelwrapper.py
+++ b/tests/test_modelwrapper.py
@@ -86,6 +86,7 @@ def test_modelwrapper():
     lfc.load_state_dict(checkpoint["state_dict"])
     bo.export_finn_onnx(lfc, (1, 1, 28, 28), export_onnx_path)
     model = ModelWrapper(export_onnx_path)
+    assert model.check_all_tensor_shapes_specified() is False
     inp_shape = model.get_tensor_shape("0")
     assert inp_shape == [1, 1, 28, 28]
     l0_weights = model.get_initializer("26")