diff --git a/llmfoundry/registry.py b/llmfoundry/registry.py
index 5924070497..50481211ac 100644
--- a/llmfoundry/registry.py
+++ b/llmfoundry/registry.py
@@ -83,7 +83,8 @@
     """
 )
 callbacks_with_config = create_registry(
-    'llm_foundry.callbacks_with_config',
+    'llmfoundry',
+    'callbacks_with_config',
     generic_type=Type[CallbackWithConfig],
     entry_points=True,
     description=_callbacks_with_config_description,