stage2 now works; tests pass

1 files changed, 4 insertions, 4 deletions

diff --git a/nizk/stage1_test.go b/nizk/stage1_test.go
index d2e4fd1..d17956d 100644
--- a/nizk/stage1_test.go
+++ b/nizk/stage1_test.go
@@ -13,8 +13,8 @@ func TestStage1Simple(t *testing.T) {
 
 	c1 := b1.StageCommit()
 	c2 := b2.StageCommit()
-	r1, pr1 := b1.RevealStage1() // Note: no Xs.
-	r2, pr2 := b2.RevealStage1() // Note: no Xs.
+	r1, pr1 := b1.RevealStage1(c1.X, c2.X)
+	r2, pr2 := b2.RevealStage1(c2.X)
 	if !b1.Commitment.VerifyStage1(c1, r1, pr1) {
 		t.Fatal("Could not verify st1 with c1 and pr1, plus=true case")
 	}
@@ -38,8 +38,8 @@ func TestStage1FromScalars(t *testing.T) {
 
 	c1 := b1.StageFromScalars(r, x)
 	c2 := b2.StageFromScalars(x, r)
-	r1, pr1 := b1.RevealStage1() // Note: no Xs
-	r2, pr2 := b2.RevealStage1() // Note: no Xs
+	r1, pr1 := b1.RevealStage1(c1.X)
+	r2, pr2 := b2.RevealStage1(c2.X)
 	if !b1.Commitment.VerifyStage1(c1, r1, pr1) {
 		t.Fatal("Could not verify st1 with c1 and pr1, plus=true case")
 	}