diff --git a/src/commonTest/kotlin/AnthropicTest.kt b/src/commonTest/kotlin/AnthropicTest.kt
index 392df28..50af56b 100644
--- a/src/commonTest/kotlin/AnthropicTest.kt
+++ b/src/commonTest/kotlin/AnthropicTest.kt
@@ -16,6 +16,7 @@ import com.xemantic.anthropic.content.Text
 import com.xemantic.anthropic.content.ToolUse
 import com.xemantic.anthropic.usage.Cost
 import com.xemantic.anthropic.usage.Usage
+import com.xemantic.kotlin.test.assert
 import com.xemantic.kotlin.test.be
 import com.xemantic.kotlin.test.have
 import com.xemantic.kotlin.test.should
diff --git a/src/commonTest/kotlin/content/DocumentTest.kt b/src/commonTest/kotlin/content/DocumentTest.kt
index b99a14d..2252fc4 100644
--- a/src/commonTest/kotlin/content/DocumentTest.kt
+++ b/src/commonTest/kotlin/content/DocumentTest.kt
@@ -3,6 +3,7 @@ package com.xemantic.anthropic.content
 import com.xemantic.anthropic.Anthropic
 import com.xemantic.anthropic.message.Message
 import com.xemantic.anthropic.message.StopReason
+import com.xemantic.kotlin.test.assert
 import com.xemantic.kotlin.test.be
 import com.xemantic.kotlin.test.have
 import com.xemantic.kotlin.test.should