diff --git a/sm4/arm64_verify_test.go b/sm4/arm64_verify_test.go
index 58a8c9a..40d6a1f 100644
--- a/sm4/arm64_verify_test.go
+++ b/sm4/arm64_verify_test.go
@@ -1,6 +1,13 @@
 //go:build arm64
 // +build arm64
 
+package sm4
+
+import (
+	"reflect"
+	"testing"
+)
+
 //go:noescape
 func tblAsm(in, imm, out *byte)
 
@@ -13,4 +20,4 @@ func TestTblAsm(t *testing.T) {
 	if !reflect.DeepEqual(out, imm) {
 		t.Errorf("expected=%v, result=%v\n", imm, out)
 	}
-}
\ No newline at end of file
+}