Fixes user module namespace for custom types

diff --git a/alembic/autogenerate/render.py b/alembic/autogenerate/render.py
index 16f8c29f9a8ac657dcc6bb9ecb72fbe7fe4b82b2..caa36d9698f698c5f3d32d37ee20f5b52a6216fd 100644 (file)
--- a/alembic/autogenerate/render.py
+++ b/alembic/autogenerate/render.py
@@ -500,7 +500,7 @@ def _repr_type(type_, autogen_context):
         if imports is not None:
             imports.add("from sqlalchemy.dialects import %s" % dname)
         return "%s.%r" % (dname, type_)
-    elif mod.startswith("sqlalchemy"):
+    elif mod.startswith("sqlalchemy."):
         prefix = _sqlalchemy_autogenerate_prefix(autogen_context)
         return "%s%r" % (prefix, type_)
     else: