diff --git a/keras_hub/src/models/task.py b/keras_hub/src/models/task.py
index b107284444..5bb954beab 100644
--- a/keras_hub/src/models/task.py
+++ b/keras_hub/src/models/task.py
@@ -339,7 +339,7 @@ def add_layer(layer, info):
                         add_layer(layer, info)
                     elif isinstance(layer, ImageConverter):
                         info = "Image size: "
-                        info += highlight_shape(layer.image_size())
+                        info += highlight_shape(layer.image_size)
                         add_layer(layer, info)
                     elif isinstance(layer, AudioConverter):
                         info = "Audio shape: "