diff --git a/hydra_storm/engine.py b/hydra_storm/engine.py
index 2989d905f..8037746ff 100644
--- a/hydra_storm/engine.py
+++ b/hydra_storm/engine.py
@@ -26,7 +26,7 @@ class StormHydraRenderEngine(bpy.types.HydraRenderEngine):
             'maxLights': settings.max_lights,
         }
 
-        if engine_type == 'FINAL':
+        if engine_type != 'VIEWPORT':
             result |= {
                 'aovToken:Combined': "color",
                 'aovToken:Depth': "depth",