diff --git a/MP1/generate_log.py b/MP1/generate_log.py
index 626e537..9758f0d 100644
--- a/MP1/generate_log.py
+++ b/MP1/generate_log.py
@@ -1,4 +1,5 @@
 import random
+import sys
 
 machine_num = int(sys.argv[1])
 machine_log_file = f"machine.{machine_num}.test.log"