From 71657b8d344a19792902820f378079974b8ca353 Mon Sep 17 00:00:00 2001
From: Sun Yimin <emmansun@users.noreply.github.com>
Date: Tue, 7 Mar 2023 08:55:09 +0800
Subject: [PATCH] fix possible test failure

---
 internal/sm2ec/sm2p256_mqv_test.go | 2 +-
 sm9/bn256/g1.go                    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/sm2ec/sm2p256_mqv_test.go b/internal/sm2ec/sm2p256_mqv_test.go
index c790975..3a2b6e0 100644
--- a/internal/sm2ec/sm2p256_mqv_test.go
+++ b/internal/sm2ec/sm2p256_mqv_test.go
@@ -14,7 +14,7 @@ import (
 func randomK(r io.Reader, ord *big.Int) (k *big.Int, err error) {
 	for {
 		k, err = rand.Int(r, ord)
-		if k.Sign() > 0 || err != nil {
+		if err != nil || (k.Sign() > 0 && len(k.Bytes()) == 32) {
 			return
 		}
 	}
diff --git a/sm9/bn256/g1.go b/sm9/bn256/g1.go
index 7e0feed..a1777aa 100644
--- a/sm9/bn256/g1.go
+++ b/sm9/bn256/g1.go
@@ -12,7 +12,7 @@ import (
 func randomK(r io.Reader) (k *big.Int, err error) {
 	for {
 		k, err = rand.Int(r, Order)
-		if k.Sign() > 0 || err != nil {
+		if err != nil || k.Sign() > 0 {
 			return
 		}
 	}