From 40aaa7d552726debe819456892e3587ffab018c8 Mon Sep 17 00:00:00 2001
From: jalving <jhjalving@gmail.com>
Date: Fri, 2 Feb 2024 14:14:39 -0800
Subject: [PATCH] fix operator

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

diff --git a/src/optiedge.jl b/src/optiedge.jl
index da846c3..e942002 100644
--- a/src/optiedge.jl
+++ b/src/optiedge.jl
@@ -192,10 +192,10 @@ function _extract_variables(func::JuMP.GenericNonlinearExpr)
     vars = NodeVariableRef[]
     for i = 1:length(func.args)
         func_arg = func.args[i]
-        if func_arg isa JuMP.GenericNonlinearExpr
-            append!(vars, _extract_variables(func_arg))
-        elseif typeof(func_arg) == NodeVariableRef
+        if typeof(func_arg) == NodeVariableRef #
             push!(vars, func_arg)
+        else
+            append!(vars, _extract_variables(func_arg))
         end
     end
     return vars