1 files changed, 4 insertions, 3 deletions
diff --git a/BackEnd/Timeline/Services/HighlightTimelineService.cs b/BackEnd/Timeline/Services/HighlightTimelineService.cs
index 88ad4a4b..619bc33e 100644
--- a/BackEnd/Timeline/Services/HighlightTimelineService.cs
+++ b/BackEnd/Timeline/Services/HighlightTimelineService.cs
@@ -4,6 +4,7 @@ using System.Collections.Generic;
 using System.Linq;
 using System.Threading.Tasks;
 using Timeline.Entities;
+using Timeline.Models;
 using Timeline.Services.Exceptions;
 
 namespace Timeline.Services
@@ -14,7 +15,7 @@ namespace Timeline.Services
         /// Get all highlight timelines.
         /// </summary>
         /// <returns>A list of all highlight timelines.</returns>
-        Task<List<Models.Timeline>> GetHighlightTimelines();
+        Task<List<TimelineInfo>> GetHighlightTimelines();
 
         /// <summary>
         /// Add a timeline to highlight list.
@@ -73,11 +74,11 @@ namespace Timeline.Services
             await _database.SaveChangesAsync();
         }
 
-        public async Task<List<Models.Timeline>> GetHighlightTimelines()
+        public async Task<List<TimelineInfo>> GetHighlightTimelines()
         {
             var entities = await _database.HighlightTimelines.Select(t => new { t.Id }).ToListAsync();
 
-            var result = new List<Models.Timeline>();
+            var result = new List<TimelineInfo>();
 
             foreach (var entity in entities)
             {