diff --git a/drugforge-alchemy/drugforge/alchemy/schema/_util.py b/drugforge-alchemy/drugforge/alchemy/schema/_util.py
index cb4de2ab..b2a75c58 100644
--- a/drugforge-alchemy/drugforge/alchemy/schema/_util.py
+++ b/drugforge-alchemy/drugforge/alchemy/schema/_util.py
@@ -1,6 +1,6 @@
 from collections import Counter
 from typing import TYPE_CHECKING
-from gufe.custom_json import JSONCodec
+from gufe.serialization.json import JSONCodec
 from alchemiscale import ScopedKey
 
 if TYPE_CHECKING: