diff --git a/app/app.py b/app/app.py
index d923f93..1f0e2bd 100644
--- a/app/app.py
+++ b/app/app.py
@@ -85,7 +85,7 @@ def format_timestamp(seconds: float, always_include_hours: bool = False, decimal
     start = time.time()
     random_inputs = {
         "input_features": np.ones(
-            BATCH_SIZE, pipeline.model.config.num_mel_bins, 2 * pipeline.model.config.max_source_positions
+            (BATCH_SIZE, pipeline.model.config.num_mel_bins, 2 * pipeline.model.config.max_source_positions)
         )
     }
     random_timestamps = pipeline.forward(random_inputs, batch_size=BATCH_SIZE, return_timestamps=True)