2 files changed, 5 insertions, 1 deletions
diff --git a/animism-align/cli/app/sql/models/annotation.py b/animism-align/cli/app/sql/models/annotation.py
index 6cc476c..cc53bd6 100644
--- a/animism-align/cli/app/sql/models/annotation.py
+++ b/animism-align/cli/app/sql/models/annotation.py
@@ -27,7 +27,7 @@ class Annotation(Base):
       'paragraph_id': self.paragraph_id,
       'start_ts': self.start_ts,
       'end_ts': self.end_ts,
-      'sentence': self.description,
+      'text': self.text,
       'settings': self.settings,
     }
 
diff --git a/animism-align/cli/app/sql/models/paragraph.py b/animism-align/cli/app/sql/models/paragraph.py
index 7c7bcd7..790c9f0 100644
--- a/animism-align/cli/app/sql/models/paragraph.py
+++ b/animism-align/cli/app/sql/models/paragraph.py
@@ -14,12 +14,16 @@ class Paragraph(Base):
   __tablename__ = 'paragraph'
   id = Column(Integer, primary_key=True)
   type = Column(String(16, convert_unicode=True), nullable=False)
+  start_ts = Column(Float, nullable=False)
+  end_ts = Column(Float, nullable=True)
   settings = Column(JSON, default={}, nullable=True)
 
   def toJSON(self):
     return {
       'id': self.id,
       'type': self.type,
+      'start_ts': self.start_ts,
+      'end_ts': self.end_ts,
       'settings': self.settings,
     }