diff --git a/create_submission.py b/create_submission.py
index 86f6bd846fdbcbd0058ec3c5e1dd9d7b55d3aeae..45aebaa5f7c5abf4759c81ba643463e3cdba50f4 100644
--- a/create_submission.py
+++ b/create_submission.py
@@ -23,7 +23,7 @@ import torch
 # Local imports
 import bindataset as dataset
 
-def create_submission(model):
+def create_submission(model, transform):
     step_days = 10
     batch_size = 1024
     # We make chunks of num_days consecutive samples; As our dummy predictor
diff --git a/main.py b/main.py
index 1d844c5ebd3f84a386f7ac8315d288697c2b7f74..a29d8dc042daad917efac0f5f85c721257e368e4 100644
--- a/main.py
+++ b/main.py
@@ -79,7 +79,7 @@ if __name__ == "__main__":
         val_loss = test.test(model, valid_loader, f_loss, device)
         print(" Validation : Loss : {:.4f}".format(val_loss))
 
-    create_submission.create_submission(model)
+    create_submission.create_submission(model, None)
     """
     logdir = generate_unique_logpath(top_logdir, "linear")
     print("Logging to {}".format(logdir))