1 files changed, 13 insertions, 13 deletions
diff --git a/nizk/stage1_test.go b/nizk/stage1_test.go
index 2dd719b..d2e4fd1 100644
--- a/nizk/stage1_test.go
+++ b/nizk/stage1_test.go
@@ -8,21 +8,21 @@ import (
 
 func TestStage1Simple(t *testing.T) {
 	id := Curve.RandomScalar()
-	b1, bc1, _ := NewBit(id, true)
-	b2, bc2, _ := NewBit(id, false)
+	b1 := NewBit(id, true)
+	b2 := NewBit(id, false)
 
 	c1 := b1.StageCommit()
 	c2 := b2.StageCommit()
 	r1, pr1 := b1.RevealStage1() // Note: no Xs.
 	r2, pr2 := b2.RevealStage1() // Note: no Xs.
-	if !bc1.VerifyStage1(c1, r1, pr1) {
+	if !b1.Commitment.VerifyStage1(c1, r1, pr1) {
 		t.Fatal("Could not verify st1 with c1 and pr1, plus=true case")
 	}
-	if !bc2.VerifyStage1(c2, r2, pr2) {
+	if !b2.Commitment.VerifyStage1(c2, r2, pr2) {
 		t.Fatal("Could not verify st2 with c2 and pr2, plus=false case")
 	}
 	// Wrong proof test
-	if bc1.VerifyStage1(c1, r1, pr2) {
+	if b1.Commitment.VerifyStage1(c1, r1, pr2) {
 		t.Fatal("Shouldn't be able to verify c1 with pr2")
 	}
 }
@@ -33,24 +33,24 @@ func TestStage1FromScalars(t *testing.T) {
 		*s = Curve.RandomScalar()
 	}
 
-	b1, bc1, _ := NewBitFromScalars(id, true, α, β)
-	b2, bc2, _ := NewBitFromScalars(id, false, α, β)
+	b1 := NewBitFromScalars(id, true, α, β)
+	b2 := NewBitFromScalars(id, false, α, β)
 
 	c1 := b1.StageFromScalars(r, x)
 	c2 := b2.StageFromScalars(x, r)
 	r1, pr1 := b1.RevealStage1() // Note: no Xs
 	r2, pr2 := b2.RevealStage1() // Note: no Xs
-	if !bc1.VerifyStage1(c1, r1, pr1) {
+	if !b1.Commitment.VerifyStage1(c1, r1, pr1) {
 		t.Fatal("Could not verify st1 with c1 and pr1, plus=true case")
 	}
-	if !bc2.VerifyStage1(c2, r2, pr2) {
+	if !b2.Commitment.VerifyStage1(c2, r2, pr2) {
 		t.Fatal("Could not verify st2 with c2 and pr2, plus=false case")
 	}
 	// Wrong proof test
-	if bc1.VerifyStage1(c1, r1, pr2) ||
-		bc1.VerifyStage1(c2, r2, pr2) ||
-		bc2.VerifyStage1(c1, r1, pr2) ||
-		bc2.VerifyStage1(c2, r2, pr1) {
+	if b2.Commitment.VerifyStage1(c1, r1, pr2) ||
+		b1.Commitment.VerifyStage1(c2, r2, pr2) ||
+		b2.Commitment.VerifyStage1(c1, r1, pr2) ||
+		b2.Commitment.VerifyStage1(c2, r2, pr1) {
 		t.Fatal("Shouldn't be able to verify bc_i with c_j or pr_j")
 	}
 }