diff --git a/knet/src/main/kotlin/com/jasonernst/knet/nextheader/ICMPNextHeaderWrapper.kt b/knet/src/main/kotlin/com/jasonernst/knet/nextheader/ICMPNextHeaderWrapper.kt
index 4775edf..eada5aa 100644
--- a/knet/src/main/kotlin/com/jasonernst/knet/nextheader/ICMPNextHeaderWrapper.kt
+++ b/knet/src/main/kotlin/com/jasonernst/knet/nextheader/ICMPNextHeaderWrapper.kt
@@ -6,13 +6,9 @@ import java.nio.ByteOrder
 class ICMPNextHeaderWrapper(
     private val icmpHeader: ICMPHeader,
     override val protocol: UByte,
-    override val typeString: String
-): NextHeader {
-    override fun getHeaderLength(): UShort {
-        return icmpHeader.size().toUShort()
-    }
+    override val typeString: String,
+) : NextHeader {
+    override fun getHeaderLength(): UShort = icmpHeader.size().toUShort()
 
-    override fun toByteArray(order: ByteOrder): ByteArray {
-        return icmpHeader.toByteArray(order)
-    }
-}
\ No newline at end of file
+    override fun toByteArray(order: ByteOrder): ByteArray = icmpHeader.toByteArray(order)
+}