diff --git a/scripts/example.py b/scripts/example.py
index 03a1c453..9132b17e 100644
--- a/scripts/example.py
+++ b/scripts/example.py
@@ -91,6 +91,7 @@
 
 # If TEST and TOKEN, submit to crowdAI
 if not args.train and args.token:
+    agent.load_weights(args.model)
     # Settings
     remote_base = 'http://grader.crowdai.org:1729'
     client = Client(remote_base)