diff --git a/.vscode/settings.json b/.vscode/settings.json
index e916afb..af5da48 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -31,7 +31,6 @@
     "Parcoords",
     "plotly",
     "proba",
-    "psutil",
     "pydantic",
     "pymongo",
     "pyplot",
diff --git a/src/great_ai/utilities/parallel_map.py b/src/great_ai/utilities/parallel_map.py
index 38a341e..14fa621 100644
--- a/src/great_ai/utilities/parallel_map.py
+++ b/src/great_ai/utilities/parallel_map.py
@@ -2,7 +2,6 @@ from math import ceil
 from typing import Any, Callable, Iterable, List, Optional
 
 import multiprocess as mp
-import psutil
 from tqdm.cli import tqdm
 
 from .logger import get_logger
@@ -18,7 +17,7 @@ def parallel_map(
     disable_progress: bool = False,
 ) -> List[Any]:
     if concurrency is None:
-        concurrency = psutil.cpu_count()
+        concurrency = mp.cpu_count()
 
     assert concurrency > 0
     assert chunk_size is None or chunk_size > 0