From e992a518bee6153af381b0f587e3a673c687cb9a Mon Sep 17 00:00:00 2001
From: odow <o.dowson@gmail.com>
Date: Wed, 9 Aug 2023 13:25:05 +1200
Subject: [PATCH] Update to latest

---
 src/MOI_wrapper.jl | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/MOI_wrapper.jl b/src/MOI_wrapper.jl
index 64dcaa9..e442959 100644
--- a/src/MOI_wrapper.jl
+++ b/src/MOI_wrapper.jl
@@ -230,8 +230,11 @@ function _F_nonlinear_operator(model::Optimizer)
         N = div(MOI.dimension(s), 2)
         for i in 1:N
             xi = f.rows[i+N]
-            @assert xi isa MOI.VariableIndex
-            f_map[xi.value] = f.rows[i]
+            # Hacky way to ensure that xi is a standalone variable
+            @assert xi isa MOI.ScalarNonlinearFunction
+            @assert xi.head == :+ && length(xi.args) == 1
+            @assert xi.args[1] isa MOI.VariableIndex
+            f_map[xi.args[1].value] = f.rows[i]
         end
     end
     nlp = MOI.Nonlinear.Model()