diff --git a/node_wrangler.py b/node_wrangler.py
index 290ac082a23a0ca9c0e01cd62ce6568f45b3bc30..cc1b85febb5065d5859e9871c248b8a5f77a685e 100644
--- a/node_wrangler.py
+++ b/node_wrangler.py
@@ -830,12 +830,13 @@ class NWNodeWrangler(bpy.types.AddonPreferences):
 
 def nw_check(context):
     space = context.space_data
-    editor_is_valid = space.type == 'NODE_EDITOR'
-
     valid_trees = ["ShaderNodeTree", "CompositorNodeTree", "TextureNodeTree", "GeometryNodeTree"]
-    tree_is_valid = space.node_tree is not None and space.tree_type in valid_trees
 
-    return editor_is_valid and tree_is_valid
+    valid = False
+    if space.type == 'NODE_EDITOR' and space.node_tree is not None and space.tree_type in valid_trees:
+        valid = True
+
+    return valid
 
 class NWBase:
     @classmethod