1 files changed, 6 insertions, 5 deletions
diff --git a/mediagoblin/media_types/video/models.py b/mediagoblin/media_types/video/models.py
index 741c329b..709d7910 100644
--- a/mediagoblin/media_types/video/models.py
+++ b/mediagoblin/media_types/video/models.py
@@ -18,16 +18,17 @@
 from mediagoblin.db.sql.models import Base
 
 from sqlalchemy import (
-    Column, Integer, Unicode, UnicodeText, DateTime, Boolean, ForeignKey,
-    UniqueConstraint)
+    Column, Integer, SmallInteger, ForeignKey)
 
 
 class VideoData(Base):
     __tablename__ = "video_data"
 
-    id = Column(Integer, primary_key=True)
-    media_entry = Column(
-        Integer, ForeignKey('media_entries.id'), nullable=False)
+    # The primary key *and* reference to the main media_entry
+    media_entry = Column(Integer, ForeignKey('media_entries.id'),
+        primary_key=True)
+    width = Column(SmallInteger)
+    height = Column(SmallInteger)
 
 
 DATA_MODEL = VideoData