From feaf52282c8513228c763707ff76d83a86444ba3 Mon Sep 17 00:00:00 2001
From: Bagaev Dmitry <bvdmitri@gmail.com>
Date: Tue, 7 May 2024 18:39:52 +0200
Subject: [PATCH] 2prev

---
 src/graph_engine.jl | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/graph_engine.jl b/src/graph_engine.jl
index 975b0fda..7f0da5f7 100644
--- a/src/graph_engine.jl
+++ b/src/graph_engine.jl
@@ -307,9 +307,12 @@ function LazyLabel(name, model, context, index::Tuple)
     return LazyLabel(name, model, context)
 end
 
-proxylabel(name::Symbol, index::Nothing, proxied::LazyLabel) = proxied
+proxylabel(name::Symbol, index::Nothing, proxied::LazyLabel) = ProxyLabel(name, index, proxied)
 proxylabel(name::Symbol, index::Tuple, proxied::LazyLabel) = ProxyLabel(name, index, proxied)
 
+proxylabel(name::Symbol, index::Any, proxied::ProxyLabel{Nothing, <:LazyLabel}) = ProxyLabel(name, index, proxied.proxied)
+proxylabel(name::Symbol, index::Nothing, proxied::ProxyLabel{Nothing, <:LazyLabel}) = ProxyLabel(name, index, proxied.proxied)
+
 function __proxy_unroll(proxied::LazyLabel)
     return haskey(proxied.context, proxied.name) ? proxied.context[proxied.name] : getorcreate!(proxied.model, proxied.context, proxied.name, nothing)
 end