diff --git a/openpype/hosts/aftereffects/api/launch_logic.py b/openpype/hosts/aftereffects/api/launch_logic.py
index c549268978..30a3e1f1c3 100644
--- a/openpype/hosts/aftereffects/api/launch_logic.py
+++ b/openpype/hosts/aftereffects/api/launch_logic.py
@@ -12,9 +12,8 @@ from wsrpc_aiohttp import (
 
 from Qt import QtCore
 
+from openpype.pipeline import legacy_io
 from openpype.tools.utils import host_tools
-
-from avalon import api
 from openpype.tools.adobe_webserver.app import WebServerTool
 
 from .ws_stub import AfterEffectsServerStub
@@ -271,13 +270,13 @@ class AfterEffectsRoute(WebSocketRoute):
         log.info("Setting context change")
         log.info("project {} asset {} ".format(project, asset))
         if project:
-            api.Session["AVALON_PROJECT"] = project
+            legacy_io.Session["AVALON_PROJECT"] = project
             os.environ["AVALON_PROJECT"] = project
         if asset:
-            api.Session["AVALON_ASSET"] = asset
+            legacy_io.Session["AVALON_ASSET"] = asset
             os.environ["AVALON_ASSET"] = asset
         if task:
-            api.Session["AVALON_TASK"] = task
+            legacy_io.Session["AVALON_TASK"] = task
             os.environ["AVALON_TASK"] = task
 
     async def read(self):
diff --git a/openpype/hosts/aftereffects/api/pipeline.py b/openpype/hosts/aftereffects/api/pipeline.py
index 3ed2de0e9d..73aea2da11 100644
--- a/openpype/hosts/aftereffects/api/pipeline.py
+++ b/openpype/hosts/aftereffects/api/pipeline.py
@@ -2,10 +2,8 @@ import os
 import sys
 
 from Qt import QtWidgets
-from bson.objectid import ObjectId
 
 import pyblish.api
-from avalon import io
 
 from openpype import lib
 from openpype.api import Logger
@@ -15,7 +13,6 @@ from openpype.pipeline import (
     deregister_loader_plugin_path,
     deregister_creator_plugin_path,
     AVALON_CONTAINER_ID,
-    registered_host,
 )
 import openpype.hosts.aftereffects
 from openpype.lib import register_event_callback
diff --git a/openpype/hosts/aftereffects/plugins/publish/collect_workfile.py b/openpype/hosts/aftereffects/plugins/publish/collect_workfile.py
index cb5a2bad4f..21a0cd7a1b 100644
--- a/openpype/hosts/aftereffects/plugins/publish/collect_workfile.py
+++ b/openpype/hosts/aftereffects/plugins/publish/collect_workfile.py
@@ -1,5 +1,5 @@
 import os
-from avalon import api
+
 import pyblish.api
 from openpype.lib import get_subset_name_with_asset_doc
 
@@ -11,7 +11,6 @@ class CollectWorkfile(pyblish.api.ContextPlugin):
     order = pyblish.api.CollectorOrder + 0.1
 
     def process(self, context):
-        task = api.Session["AVALON_TASK"]
         current_file = context.data["currentFile"]
         staging_dir = os.path.dirname(current_file)
         scene_file = os.path.basename(current_file)
diff --git a/openpype/hosts/aftereffects/plugins/publish/validate_instance_asset.py b/openpype/hosts/aftereffects/plugins/publish/validate_instance_asset.py
index 37cecfbcc4..1a303f5da4 100644
--- a/openpype/hosts/aftereffects/plugins/publish/validate_instance_asset.py
+++ b/openpype/hosts/aftereffects/plugins/publish/validate_instance_asset.py
@@ -1,7 +1,10 @@
-from avalon import api
 import pyblish.api
+
 import openpype.api
-from openpype.pipeline import PublishXmlValidationError
+from openpype.pipeline import (
+    PublishXmlValidationError,
+    legacy_io,
+)
 from openpype.hosts.aftereffects.api import get_stub
 
 
@@ -27,7 +30,7 @@ class ValidateInstanceAssetRepair(pyblish.api.Action):
         for instance in instances:
             data = stub.read(instance[0])
 
-            data["asset"] = api.Session["AVALON_ASSET"]
+            data["asset"] = legacy_io.Session["AVALON_ASSET"]
             stub.imprint(instance[0], data)
 
 
@@ -51,7 +54,7 @@ class ValidateInstanceAsset(pyblish.api.InstancePlugin):
 
     def process(self, instance):
         instance_asset = instance.data["asset"]
-        current_asset = api.Session["AVALON_ASSET"]
+        current_asset = legacy_io.Session["AVALON_ASSET"]
         msg = (
             f"Instance asset {instance_asset} is not the same "
             f"as current context {current_asset}."