update

1 files changed, 9 insertions, 2 deletions

diff --git a/script/vm/node.lua b/script/vm/node.lua
index 3368ab05..082a5e87 100644
--- a/script/vm/node.lua
+++ b/script/vm/node.lua
@@ -38,10 +38,14 @@ function m.setNode(source, node)
     m.nodeCache[source] = m.mergeNode(me, node)
 end
 
+function m.getNode(source)
+    return m.nodeCache[source]
+end
+
 ---@param node vm.node
 ---@return vm.node.union
 function m.addOptional(node)
-    if node.type ~= 'union' then
+    if not node or node.type ~= 'union' then
         node = union(node)
     end
     node = node:addOptional()
@@ -49,8 +53,11 @@ function m.addOptional(node)
 end
 
 ---@param node vm.node
----@return vm.node.union
+---@return vm.node.union?
 function m.removeOptional(node)
+    if not node then
+        return node
+    end
     if node.type ~= 'union' then
         node = union(node)
     end