2 files changed, 7 insertions, 1 deletions
diff --git a/Timeline/Entities/TimelineEntity.cs b/Timeline/Entities/TimelineEntity.cs
index 56b36d4e..3149d4c2 100644
--- a/Timeline/Entities/TimelineEntity.cs
+++ b/Timeline/Entities/TimelineEntity.cs
@@ -2,7 +2,7 @@
 using System.Collections.Generic;
 using System.ComponentModel.DataAnnotations;
 using System.ComponentModel.DataAnnotations.Schema;
-using Timeline.Models.Http;
+using Timeline.Models;
 
 namespace Timeline.Entities
 {
diff --git a/Timeline/Entities/TimelinePostEntity.cs b/Timeline/Entities/TimelinePostEntity.cs
index 5805abe0..24bfc7a3 100644
--- a/Timeline/Entities/TimelinePostEntity.cs
+++ b/Timeline/Entities/TimelinePostEntity.cs
@@ -25,9 +25,15 @@ namespace Timeline.Entities
         [ForeignKey(nameof(AuthorId))]
         public UserEntity Author { get; set; } = default!;
 
+        [Column("content_type"), Required]
+        public string ContentType { get; set; } = default!;
+
         [Column("content")]
         public string? Content { get; set; }
 
+        [Column("extra_content")]
+        public string? ExtraContent { get; set; }
+
         [Column("time")]
         public DateTime Time { get; set; }