diff --git a/android/app/src/test/kotlin/app/donottype/core/TokenUsageTest.kt b/android/app/src/test/kotlin/app/donottype/core/TokenUsageTest.kt
index b59ee15..6ddbacc 100644
--- a/android/app/src/test/kotlin/app/donottype/core/TokenUsageTest.kt
+++ b/android/app/src/test/kotlin/app/donottype/core/TokenUsageTest.kt
@@ -1,7 +1,7 @@
 package app.donottype.core
 
-import kotlin.test.assertEquals
-import kotlin.test.assertNull
+import org.junit.Assert.assertEquals
+import org.junit.Assert.assertNull
 import org.junit.Test
 
 /**
diff --git a/android/app/src/test/kotlin/app/donottype/core/TruncationGuardTest.kt b/android/app/src/test/kotlin/app/donottype/core/TruncationGuardTest.kt
index e812d2e..a5a9a5c 100644
--- a/android/app/src/test/kotlin/app/donottype/core/TruncationGuardTest.kt
+++ b/android/app/src/test/kotlin/app/donottype/core/TruncationGuardTest.kt
@@ -1,7 +1,7 @@
 package app.donottype.core
 
-import kotlin.test.assertFalse
-import kotlin.test.assertTrue
+import org.junit.Assert.assertFalse
+import org.junit.Assert.assertTrue
 import org.junit.Test
 
 /**