diff --git a/api/api_v1/endpoints/query.py b/api/api_v1/endpoints/query.py
index 4184bfb..8c8c224 100644
--- a/api/api_v1/endpoints/query.py
+++ b/api/api_v1/endpoints/query.py
@@ -152,7 +152,7 @@ async def retention_model(
     df.set_index(res['groupby'], inplace=True)
     df.sort_index(inplace=True)
     values = {}
-    days = [i for i in range((date_range[-1] - date_range[0]).days + 1)][:unit_num]
+    days = [i for i in range((date_range[-1] - date_range[0]).days + 1)][:unit_num+1]
     for i, d1 in enumerate(date_range):
         for g in groups:
             a = set(df.loc[(d1, event_a, *g)]['values']) if (d1, event_a, *g) in df.index else set()
@@ -171,7 +171,7 @@ async def retention_model(
         'summary_values': summary_values,
         'values': values,
         'days': days,
-        'date_range': [d.strftime('%Y-%m-%d') for d in date_range][:unit_num],
+        'date_range': [d.strftime('%Y-%m-%d') for d in date_range][:unit_num+1],
         'title': title
     }
     return schemas.Msg(code=0, msg='ok', data=data)