diff --git a/sm2/sm2_keyexchange.go b/sm2/sm2_keyexchange.go
index 0170cda..8958bb8 100644
--- a/sm2/sm2_keyexchange.go
+++ b/sm2/sm2_keyexchange.go
@@ -93,7 +93,7 @@ func (ke *KeyExchange) SetPeerParameters(peerPub *ecdsa.PublicKey, peerUID []byt
 		return errors.New("sm2: 'peerPub' already exists, please do not set it")
 	}
 
-	if !IsSM2PublicKey(peerPub) {
+	if peerPub.Curve != ke.privateKey.Curve {
 		return errors.New("sm2: peer public key is not expected/supported")
 	}