diff --git a/element_deeplabcut/train.py b/element_deeplabcut/train.py
index 6c20ce9..87b7e7c 100644
--- a/element_deeplabcut/train.py
+++ b/element_deeplabcut/train.py
@@ -106,6 +106,8 @@ class File(dj.Part):
         definition = """
         # Paths of training files (e.g., labeled pngs, CSV or video)
         -> master
+        file_id: int
+        ---
         file_path: varchar(255)
         """