diff --git a/StaffMod/src/main/kotlin/opekope2/avm_staff/content/DataComponentTypes.kt b/StaffMod/src/main/kotlin/opekope2/avm_staff/content/DataComponentTypes.kt
index e58e70d13..0348adbf1 100644
--- a/StaffMod/src/main/kotlin/opekope2/avm_staff/content/DataComponentTypes.kt
+++ b/StaffMod/src/main/kotlin/opekope2/avm_staff/content/DataComponentTypes.kt
@@ -71,6 +71,7 @@ object DataComponentTypes : RegistryUtil<ComponentType<*>>(MOD_ID, RegistryKeys.
     @JvmField
     val ROCKET_MODE = register("rocket_mode") {
         ComponentType.builder<MinecraftUnit>()
+            .codec(MinecraftUnit.CODEC)
             .packetCodec(PacketCodec.unit(minecraftUnit))
             .build()
     }