diff --git a/image_download.py b/image_download.py
index 2d5f0b4..3e68351 100644
--- a/image_download.py
+++ b/image_download.py
@@ -23,7 +23,7 @@ def main(args):
         print(f"Creating missing folder: {args.output_folder}")
         os.mkdir(args.output_folder)
     
-    if args.from_checkpoint:
+    if args.from_checkpoint and os.path.exists("./state.pickle"):
         with open("state.pickle", "rb") as state:
             state_dict = pickle.load(state)
             offset = state_dict["row_index"]