diff --git a/src/main/java/cz/muni/fi/pv168/project/db/categorytime/CategoryTimeDao.java b/src/main/java/cz/muni/fi/pv168/project/db/categorytime/CategoryTimeDao.java
index c7917eb7ffe232227bae61cb36c0d98072f1d45a..ec3c75287ad2776e066032eabef7d7bb58710da2 100644
--- a/src/main/java/cz/muni/fi/pv168/project/db/categorytime/CategoryTimeDao.java
+++ b/src/main/java/cz/muni/fi/pv168/project/db/categorytime/CategoryTimeDao.java
@@ -100,6 +100,7 @@ public class CategoryTimeDao implements OneToNDao<CategoryTime> {
             st.setLong(1, entity.getTimeInHours());
             st.setDate(2, Date.valueOf(entity.getCompletionDate()));
             st.setLong(3, entity.getTaskId());
+            st.setLong(4, entity.getId());
 
             int rowsUpdated = st.executeUpdate();
             if (rowsUpdated == 0) {