diff --git a/api/models/workflow.py b/api/models/workflow.py
index 6a86251216..7da09a5a1b 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -787,7 +787,7 @@ class WorkflowNodeExecutionModel(Base):  # This model is expected to have `offlo
 
     __tablename__ = "workflow_node_executions"
 
-    @declared_attr
+    @declared_attr.directive
     @classmethod
     def __table_args__(cls) -> Any:
         return (