From 2ffa34e6733363695ff303d52fe8c987d1d6bf67 Mon Sep 17 00:00:00 2001
From: Alex Page <alex@anpage.me>
Date: Wed, 28 Jun 2023 03:38:10 -0400
Subject: [PATCH] Actually use result of mod_sqrt

---
 umskt/src/bink2002.rs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/umskt/src/bink2002.rs b/umskt/src/bink2002.rs
index b5cea35..8e182b9 100644
--- a/umskt/src/bink2002.rs
+++ b/umskt/src/bink2002.rs
@@ -186,9 +186,12 @@ impl ProductKey {
 
             s = &s + &c;
 
-            if mod_sqrt(&s, gen_order).is_none() {
-                no_square = true;
-            };
+            match mod_sqrt(&s, gen_order) {
+                Some(res) => s = res,
+                None => {
+                    no_square = true;
+                }
+            }
 
             s = (s - e).mod_floor(gen_order);