diff --git a/main.py b/main.py
index 9581f27..d280d9e 100644
--- a/main.py
+++ b/main.py
@@ -190,9 +190,9 @@ def main_test():
         c_pred, s_pred = pred
     elif args.model_output == "client":
         c_pred = pred
-        s_pred = np.array()
+        s_pred = np.zeros(0)
     else:
-        c_pred = np.array()
+        c_pred = np.zeros(0)
         s_pred = pred
     dataset.save_predictions(args.future_prediction, c_pred, s_pred)