summaryrefslogtreecommitdiff
path: root/alembic/context.py
diff options
context:
space:
mode:
Diffstat (limited to 'alembic/context.py')
-rw-r--r--alembic/context.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/alembic/context.py b/alembic/context.py
index 9c0f676..758fca8 100644
--- a/alembic/context.py
+++ b/alembic/context.py
@@ -1,6 +1,5 @@
-from .environment import EnvironmentContext
-from . import util
+from .runtime.environment import EnvironmentContext
# create proxy functions for
# each method on the EnvironmentContext class.
-util.create_module_class_proxy(EnvironmentContext, globals(), locals())
+EnvironmentContext.create_module_class_proxy(globals(), locals())