Fix bug from accessing last entry of new category

diff --git a/src/time_sheet.py b/src/time_sheet.py
index cbdccb16ec3972a69aea2bd9cb48a968549bf920..34686eb08480631dd273ef75e241a06b7f20f192 100644 (file)
--- a/src/time_sheet.py
+++ b/src/time_sheet.py
@@ -76,8 +76,8 @@ class TimeSheet(Puncher):
 
 
     def get_last_entry(self, cat_id: str) -> Optional[Entry]:
 
 
     def get_last_entry(self, cat_id: str) -> Optional[Entry]:

-        entries = self.cat_entries[cat_id]
-        if len(entries) == 0:
+        entries = self.cat_entries.get(cat_id)
+        if entries is None or len(entries) == 0:

             return None
 
         return self.entries[entries[-1]]
             return None
 
         return self.entries[entries[-1]]