diff --git a/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt b/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt
index 2834bcfc4..8969e414f 100644
--- a/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt
+++ b/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt
@@ -120,11 +120,10 @@ class DateUtilsTest : BaseUnitTest() {
 
     @Test
     fun getWeekdaysInMonth() {
-        fun getCalendarUTC(year: Int, month: Int, dayOfMonth: Int): GregorianCalendar {
-            val gregCalendar = GregorianCalendar(year, month, dayOfMonth)
-            gregCalendar.timeZone = TimeZone.getTimeZone("UTC")
-            return gregCalendar
-        }
+        fun getCalendarUTC(year: Int, month: Int, dayOfMonth: Int) =
+            GregorianCalendar(year, month, dayOfMonth).apply {
+                timeZone = TimeZone.getTimeZone("UTC")
+            }
 
         val february = getCalendarUTC(2018, Calendar.FEBRUARY, 1)
         val leapFebruary = getCalendarUTC(2020, Calendar.FEBRUARY, 1)