diff --git a/openpype/hosts/aftereffects/plugins/publish/extract_save_scene.py b/openpype/hosts/aftereffects/plugins/publish/extract_save_scene.py
index e20598b311..eb2977309f 100644
--- a/openpype/hosts/aftereffects/plugins/publish/extract_save_scene.py
+++ b/openpype/hosts/aftereffects/plugins/publish/extract_save_scene.py
@@ -1,15 +1,16 @@
+import pyblish.api
+
 import openpype.api
 from openpype.hosts.aftereffects.api import get_stub
 
 
-class ExtractSaveScene(openpype.api.Extractor):
+class ExtractSaveScene(pyblish.api.ContextPlugin):
     """Save scene before extraction."""
 
     order = openpype.api.Extractor.order - 0.48
     label = "Extract Save Scene"
     hosts = ["aftereffects"]
-    families = ["workfile"]
 
-    def process(self, instance):
+    def process(self, context):
         stub = get_stub()
         stub.save()