diff --git a/src/aiida_workgraph/config.py b/src/aiida_workgraph/config.py
index ab316c38..335a9ec7 100644
--- a/src/aiida_workgraph/config.py
+++ b/src/aiida_workgraph/config.py
@@ -1,5 +1,5 @@
 import json
-from aiida.manage.configuration.settings import AIIDA_CONFIG_FOLDER
+from aiida.manage.configuration.settings import AiiDAConfigDir
 
 WORKGRAPH_EXTRA_KEY = "_workgraph"
 WORKGRAPH_SHORT_EXTRA_KEY = "_workgraph_short"
@@ -11,7 +11,7 @@
 
 def load_config() -> dict:
     """Load the configuration from the config file."""
-    config_file_path = AIIDA_CONFIG_FOLDER / "workgraph.json"
+    config_file_path = AiiDAConfigDir.get() / "workgraph.json"
     try:
         with config_file_path.open("r") as f:
             config = json.load(f)