diff --git a/scan_to_paperless/status.py b/scan_to_paperless/status.py
index e7e77cf6..b732b4b0 100644
--- a/scan_to_paperless/status.py
+++ b/scan_to_paperless/status.py
@@ -272,7 +272,9 @@ def _update_status(self, name: str) -> None:
             self.set_status(name, nb_images, _WAITING_TO_STATUS.format(run_step["name"]), step=run_step)
         else:
             len_folder = len(os.path.join(source_folder, name).rstrip("/")) + 1
-            source_images = config["steps"][-2]["sources"] if len(config["steps"]) >= 2 else config["sources"]
+            source_images = (
+                config["steps"][-2]["sources"] if len(config.get("steps", [])) >= 2 else config["images"]
+            )
             generated_images = [f[len_folder:] for f in config["steps"][-1]["sources"]]
             self.set_status(
                 name,