diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py
index e3b24ba4..d885365d 100644
--- a/atst/routes/task_orders/new.py
+++ b/atst/routes/task_orders/new.py
@@ -246,8 +246,7 @@ def new(screen, task_order_id=None, portfolio_id=None):
             )
 
     if task_order_id and screen is 4:
-        task_order = TaskOrders.get(g.current_user, task_order_id)
-        if not TaskOrders.all_sections_complete(task_order):
+        if not TaskOrders.all_sections_complete(workflow.task_order):
             flash("task_order_draft")
 
     url_args = {"screen": screen}