diff --git a/uhabits-core/src/jvmMain/java/org/isoron/uhabits/core/ui/widgets/WidgetBehavior.kt b/uhabits-core/src/jvmMain/java/org/isoron/uhabits/core/ui/widgets/WidgetBehavior.kt
index 99bad0e84..2350bde85 100644
--- a/uhabits-core/src/jvmMain/java/org/isoron/uhabits/core/ui/widgets/WidgetBehavior.kt
+++ b/uhabits-core/src/jvmMain/java/org/isoron/uhabits/core/ui/widgets/WidgetBehavior.kt
@@ -46,8 +46,8 @@ class WidgetBehavior @Inject constructor(
         setValue(habit, timestamp, Entry.NO)
     }
 
-    fun onToggleRepetition(habit: Habit, timestamp: Timestamp?) {
-        val currentValue = habit.originalEntries.get(timestamp!!).value
+    fun onToggleRepetition(habit: Habit, timestamp: Timestamp) {
+        val currentValue = habit.originalEntries.get(timestamp).value
         val newValue: Int
         newValue =
             if (preferences.isSkipEnabled) nextToggleValueWithSkip(