diff --git a/CHANGELOG.md b/CHANGELOG.md
index 3e9650984..b9c83e9ea 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -35,11 +35,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - Documentation and test cases for database triggers.
+- New Runner and executor API to bypass server-side memory when running tasks.
 
 ### Docs
 
 - Added federated learning showcase code
-- Updated tutorial for redispatching workflows with Streamlit 
+- Updated tutorial for redispatching workflows with Streamlit
 
 ### Tests
 
@@ -152,7 +153,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fix for double locking file in configurations.
 - Introduced new data access layer
 - Introduced Shutil file transfer strategy for local file transfers
-- New Runner and executor API to bypass server-side memory when running tasks.
 
 ### Docs
 
diff --git a/covalent_dispatcher/_core/runner.py b/covalent_dispatcher/_core/runner.py
index 473ea0c40..08b990cd6 100644
--- a/covalent_dispatcher/_core/runner.py
+++ b/covalent_dispatcher/_core/runner.py
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 """
-Defines the core functionality of the runner
+Defines the core functionality of the legacy runner
 """
 
 import asyncio
diff --git a/covalent_dispatcher/_core/runner_ng.py b/covalent_dispatcher/_core/runner_ng.py
index b7741b80e..0ab14c3e0 100644
--- a/covalent_dispatcher/_core/runner_ng.py
+++ b/covalent_dispatcher/_core/runner_ng.py
@@ -1,4 +1,4 @@
-# Copyright 2021 Agnostiq Inc.
+# Copyright 2023 Agnostiq Inc.
 #
 # This file is part of Covalent.
 #
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 """
-Defines the core functionality of the runner
+Defines the core functionality of the new improved runner
 """
 
 import asyncio
@@ -72,6 +72,7 @@ async def _submit_abstract_task_group(
 ) -> None:
     # Task sequence of the form {"function_id": task_id, "args_ids":
     # [node_ids], "kwargs_ids": {key: node_id}}
+
     task_ids = [task["function_id"] for task in task_seq]
     task_specs = []
     task_group_metadata = {
diff --git a/tests/covalent_tests/executor/base_test.py b/tests/covalent_tests/executor/base_test.py
index 81a9d7f9f..5c6e58566 100644
--- a/tests/covalent_tests/executor/base_test.py
+++ b/tests/covalent_tests/executor/base_test.py
@@ -26,9 +26,10 @@
 from covalent import DepsCall, TransportableObject
 from covalent._results_manager import Result
 from covalent._shared_files.exceptions import TaskCancelledError, TaskRuntimeError
-from covalent.executor import BaseExecutor, wrapper_fn
+from covalent.executor import BaseExecutor
 from covalent.executor.base import AsyncBaseExecutor
 from covalent.executor.utils.enums import Signals
+from covalent.executor.utils.wrappers import wrapper_fn
 
 
 class MockExecutor(BaseExecutor):