diff options
-rw-r--r-- | node_wrangler.py | 7 |
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') # |