X-Git-Url: https://git.distorted.org.uk/u/mdw/catacomb/blobdiff_plain/cf76bcbb2097c36912fae1c8a6cc77f0ce4bfafd..191f16d5ddd637907ec076b11344512d1887a03a:/pgen-safe.c

diff --git a/pgen-safe.c b/pgen-safe.c
index 115f7b4..bd644dc 100644
--- a/pgen-safe.c
+++ b/pgen-safe.c
@@ -180,10 +180,12 @@ int pgen_safetest(int rq, pgen_event *ev, void *p)
       rc = PGEN_TRY;
       break;
     case PGEN_TRY: {
-      mp *m = mprand_range(MP_NEW, c->c.q.m, ev->r, 0);
+      mp *m = mprand_range(MP_NEW, c->c.p.m, ev->r, 0);
       rc = rabin_test(&c->p, m);
-      if (rc == PGEN_PASS)
+      if (rc == PGEN_PASS) {
+	m = mprand_range(m, c->c.q.m, ev->r, 0);
 	rc = rabin_test(&c->q, m);
+      }
       mp_drop(m);
     } break;
     case PGEN_DONE: