diff --git a/internal/cpu/cpu_s390x_test.go b/internal/cpu/cpu_s390x_test.go
index 9bee163..686e819 100644
--- a/internal/cpu/cpu_s390x_test.go
+++ b/internal/cpu/cpu_s390x_test.go
@@ -9,7 +9,7 @@ import (
 	"testing"
 	"unsafe"
 
-	"golang.org/x/sys/cpu"
+	"github.com/emmansun/gmsm/internal/cpu"
 )
 
 var s390xTests = []struct {
diff --git a/internal/cpu/endian_test.go b/internal/cpu/endian_test.go
index b066bfb..6fc68d5 100644
--- a/internal/cpu/endian_test.go
+++ b/internal/cpu/endian_test.go
@@ -8,7 +8,7 @@ import (
 	"testing"
 	"unsafe"
 
-	"golang.org/x/sys/cpu"
+	"github.com/emmansun/gmsm/internal/cpu"
 )
 
 func TestIsBigEndian(t *testing.T) {