diff --git a/examples/src/test/kotlin/CodecTest.kt b/examples/src/test/kotlin/CodecTest.kt
index b2009280..5a926b83 100644
--- a/examples/src/test/kotlin/CodecTest.kt
+++ b/examples/src/test/kotlin/CodecTest.kt
@@ -66,7 +66,7 @@ internal class CodecTest {
 
         init {
             powerStatusCodec = registry[PowerStatus::class.java]
-            integerCodec = registry.get(Int::class.java)
+            integerCodec = IntegerCodec()
         }
 
         override fun encode(writer: BsonWriter, value: Monolight, encoderContext: EncoderContext) {
diff --git a/source/examples/generated/CodecTest.snippet.example-codec-2.kt b/source/examples/generated/CodecTest.snippet.example-codec-2.kt
index f044d0b0..14352f07 100644
--- a/source/examples/generated/CodecTest.snippet.example-codec-2.kt
+++ b/source/examples/generated/CodecTest.snippet.example-codec-2.kt
@@ -4,7 +4,7 @@ class MonolightCodec(registry: CodecRegistry) : Codec<Monolight> {
 
     init {
         powerStatusCodec = registry[PowerStatus::class.java]
-        integerCodec = registry.get(Int::class.java)
+        integerCodec = IntegerCodec()
     }
 
     override fun encode(writer: BsonWriter, value: Monolight, encoderContext: EncoderContext) {