diff --git a/node_wrangler.py b/node_wrangler.py
index 48b153b20bd21ac30d9b2dcbb4bef073848af3be..1f03ecd5082b8fb3d3ed2ebc30d37f4004682daf 100644
--- a/node_wrangler.py
+++ b/node_wrangler.py
@@ -1058,8 +1058,10 @@ class NWNodeWrangler(bpy.types.AddonPreferences):
 
 def nw_check(context):
     space = context.space_data
+    valid_trees = ["ShaderNodeTree", "CompositorNodeTree", "TextureNodeTree"]
+    
     valid = False
-    if space.type == 'NODE_EDITOR' and space.node_tree is not None:
+    if space.type == 'NODE_EDITOR' and space.node_tree is not None and space.tree_type in valid_trees:
         valid = True
         
     return valid