From 42466936a3aa18f2b1f4bc104903014fcbe9b01a Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Tue, 6 Feb 2024 16:21:04 -0600
Subject: [PATCH] Minor fixes

---
 src/solvers/jump.jl | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/solvers/jump.jl b/src/solvers/jump.jl
index 448180e..d4d88bf 100644
--- a/src/solvers/jump.jl
+++ b/src/solvers/jump.jl
@@ -344,10 +344,10 @@ function _optimize(model::JuMP.Model)
     end
 
     # Optimize
-    ext.where = :WHERE_DEFAULT
     optimize!(model)
 
     # Cleanup
+    ext.where = :WHERE_DEFAULT
     ext.cb_data = nothing
     flush(stdout)
     Libc.flush_cstdio()
@@ -446,13 +446,13 @@ function __init_solvers_jump__()
             self.inner.ext[:miplearn].aot_cuts = cuts
         end
 
-        function lazy_enforce(self, model, violations)
+        function lazy_enforce(self, violations)
             self.inner.ext[:miplearn].lazy_enforce(violations)
         end
 
         function _lazy_enforce_collected(self)
             ext = self.inner.ext[:miplearn]
-            if ext.lazy_enforce !== nothing
+            if ext.lazy !== nothing
                 ext.lazy_enforce(ext.lazy)
             end
         end