diff --git a/dags/taskflow.py b/dags/taskflow.py
index 24d567297499301eeae07b549ef605cfd35babeb..c346559d1087fc9a5765cdcd14cf4604d2f12185 100644
--- a/dags/taskflow.py
+++ b/dags/taskflow.py
@@ -39,7 +39,7 @@ def taskflow_example():
         print(f"Total files downloaded: {len(files)}")
 
     data = extract(oid = 'b38609df2b334ea296ea1857e568dbea')
-    summary = transform(data)
-    load(summary["keys"])
+    files = transform(data)
+    load(files)
 
 dag = taskflow_example()