X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Ffunction%2FFunctions.java;h=cc9b336f69a796680b8fc8bd1c972900f9d35ece;hb=04d29917ceb4e34d2d9fc32ac4c7c8bdd2f4a732;hp=3aa6f2752cc04363acf417762d1e1ab680ea3ac7;hpb=03368dc8b2b09ef54172c8c9cc3724dc48f0d558;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/function/Functions.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/function/Functions.java
index 3aa6f275..cc9b336f 100644
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/function/Functions.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/function/Functions.java
@@ -601,10 +601,16 @@ public class Functions {
                 importEntry = graph.getPossibleURI(sclmain);
             }
         }
-        return new BrightnessExpressionValidator(
-                importEntry != null
-                ? Arrays.asList(importEntry)
-                : Collections.emptyList());
+        SCLContext ctx = SCLContext.getCurrent();
+        Object oldGraph = ctx.put("graph", graph);
+        try {
+            return new BrightnessExpressionValidator(
+                    importEntry != null
+                    ? Arrays.asList(importEntry)
+                    : Collections.emptyList());
+        } finally {
+            ctx.put("graph", oldGraph);
+        }
     }
 
     private static class BrightnessExpressionValidator implements Function1<String, String> {