diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py
index 14cd051..905263d 100644
--- a/ethosu/vela/high_level_command_stream_generator.py
+++ b/ethosu/vela/high_level_command_stream_generator.py
@@ -373,7 +373,7 @@
     if not passes[0].ifm_tensor or not passes[-1].ofm_tensor:
         return 0
 
-    ifm_read = passes[0].ifm_tensor.storage_size
+    ifm_read = passes[0].ifm_tensor.storage_size()
     min_overlap = 999999999999999999999
     ofm_size = passes[-1].ofm_tensor.storage_size()
     if strat == SchedulingStrategy.WeightStream: