diff --git a/zuc/eia256_asm_arm64.s b/zuc/eia256_asm_arm64.s
index a57face..a0b92a6 100644
--- a/zuc/eia256_asm_arm64.s
+++ b/zuc/eia256_asm_arm64.s
@@ -177,10 +177,10 @@ TEXT ·eia256RoundTag16(SB),NOSPLIT,$0
 	VMOV XTMP2.S[0], KS_M2.S[1]
 	VMOV XTMP2.S[2], KS_M2.S[2]
 	VMOV XTMP2.S[1], KS_M2.S[3] // KS bits [191:160 159:128 223:192 191:160]
-	VMOV XTMP2.S[3], KS_M2.S[0]
-	VMOV XTMP2.S[2], KS_M2.S[1]
-	VMOV XTMP2.S[3], KS_M2.S[2]
-	VMOV XTMP2.S[2], KS_M2.S[3] // KS bits [255:224 223:192 255:224 223:192]
+	VMOV XTMP2.S[3], KS_H.S[0]
+	VMOV XTMP2.S[2], KS_H.S[1]
+	VMOV XTMP2.S[3], KS_H.S[2]
+	VMOV XTMP2.S[2], KS_H.S[3] // KS bits [255:224 223:192 255:224 223:192]
     
 	// setup DATA
 	VTBL SHUF_MASK_DW0_DW1.B16, [XTMP3.B16], XTMP1.B16 // XTMP1 - Data bits [31:0 0s 63:32 0s]