diff --git a/src/deploy_prefect/flows.py b/src/deploy_prefect/flows.py
index c0ea92a..320b7ab 100644
--- a/src/deploy_prefect/flows.py
+++ b/src/deploy_prefect/flows.py
@@ -10,7 +10,7 @@
     upload_data_to_dune,
     update_aggregate_query,
 )
-from src.deploy_prefect.dataclasses import ENV, CHAIN, Config
+from src.deploy_prefect.models import ENV, CHAIN, Config
 
 
 @flow(retries=3, retry_delay_seconds=60, log_prints=True)  # type: ignore[misc]
diff --git a/src/deploy_prefect/tasks.py b/src/deploy_prefect/tasks.py
index 35e2b1e..e1caeed 100644
--- a/src/deploy_prefect/tasks.py
+++ b/src/deploy_prefect/tasks.py
@@ -119,7 +119,7 @@ def update_aggregate_query(table_name: str, config: Config) -> None:
             + sql_query[insertion_point:]
         )
         dune.update_query(  # type: ignore[attr-defined]
-            query_sql=updated_sql_query, query_id=dune_query_id
+            query_sql=updated_sql_query, query_id=config.dune_query_id
         )
     else:
         logger.info("Table already in query, not updating query")