diff --git a/src/amplitude_experiment/__init__.py b/src/amplitude_experiment/__init__.py
index f223f6a..cc94982 100644
--- a/src/amplitude_experiment/__init__.py
+++ b/src/amplitude_experiment/__init__.py
@@ -13,3 +13,4 @@
 from .local.client import LocalEvaluationClient
 from .local.config import LocalEvaluationConfig
 from .flagresult import FlagResult
+from .assignment import AssignmentConfig