diff --git a/api/models/workflow.py b/api/models/workflow.py
index 6e8dda429d..d851160e18 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -1769,7 +1769,7 @@ class WorkflowDraftVariable(Base):
         return self.last_edited_at is not None
 
 
-class WorkflowDraftVariableFile(Base):
+class WorkflowDraftVariableFile(TypeBase):
     """Stores metadata about files associated with large workflow draft variables.
 
     This model acts as an intermediary between WorkflowDraftVariable and UploadFile,