diff --git a/openpype/hosts/aftereffects/__init__.py b/openpype/hosts/aftereffects/__init__.py
index deae48d122..c9ad6aaeeb 100644
--- a/openpype/hosts/aftereffects/__init__.py
+++ b/openpype/hosts/aftereffects/__init__.py
@@ -1,9 +1,6 @@
-def add_implementation_envs(env, _app):
-    """Modify environments to contain all required for implementation."""
-    defaults = {
-        "OPENPYPE_LOG_NO_COLORS": "True",
-        "WEBSOCKET_URL": "ws://localhost:8097/ws/"
-    }
-    for key, value in defaults.items():
-        if not env.get(key):
-            env[key] = value
+from .module import AfterEffectsModule
+
+
+__all__ = (
+    "AfterEffectsModule",
+)
diff --git a/openpype/hosts/aftereffects/api/workio.py b/openpype/hosts/aftereffects/api/workio.py
index d6c732285a..18b40af5dc 100644
--- a/openpype/hosts/aftereffects/api/workio.py
+++ b/openpype/hosts/aftereffects/api/workio.py
@@ -1,12 +1,11 @@
 """Host API required Work Files tool"""
 import os
 
-from openpype.pipeline import HOST_WORKFILE_EXTENSIONS
 from .launch_logic import get_stub
 
 
 def file_extensions():
-    return HOST_WORKFILE_EXTENSIONS["aftereffects"]
+    return [".aep"]
 
 
 def has_unsaved_changes():
diff --git a/openpype/hosts/aftereffects/module.py b/openpype/hosts/aftereffects/module.py
new file mode 100644
index 0000000000..93d575c186
--- /dev/null
+++ b/openpype/hosts/aftereffects/module.py
@@ -0,0 +1,23 @@
+from openpype.modules import OpenPypeModule
+from openpype.modules.interfaces import IHostModule
+
+
+class AfterEffectsModule(OpenPypeModule, IHostModule):
+    name = "aftereffects"
+    host_name = "aftereffects"
+
+    def initialize(self, module_settings):
+        self.enabled = True
+
+    def add_implementation_envs(self, env, _app):
+        """Modify environments to contain all required for implementation."""
+        defaults = {
+            "OPENPYPE_LOG_NO_COLORS": "True",
+            "WEBSOCKET_URL": "ws://localhost:8097/ws/"
+        }
+        for key, value in defaults.items():
+            if not env.get(key):
+                env[key] = value
+
+    def get_workfile_extensions(self):
+        return [".aep"]