NW: fix layout errors

Forgot to check for active node and tree type.

1 files changed, 5 insertions, 2 deletions

diff --git a/node_wrangler.py b/node_wrangler.py
index 69e8b1a0..f1a42adb 100644
--- a/node_wrangler.py
+++ b/node_wrangler.py
@@ -3997,8 +3997,11 @@ def bgreset_menu_func(self, context):
 
 
 def save_viewer_menu_func(self, context):
-    if context.scene.node_tree.nodes.active.type == "VIEWER":
-        self.layout.operator(NWSaveViewer.bl_idname, icon='FILE_IMAGE')
+    if nw_check(context):
+        if context.space_data.tree_type == 'CompositorNodeTree':
+            if context.scene.node_tree.nodes.active:
+                if context.scene.node_tree.nodes.active.type == "VIEWER":
+                    self.layout.operator(NWSaveViewer.bl_idname, icon='FILE_IMAGE')
 
 
 #