diff --git a/tests/test_generates.py b/tests/test_generates.py
index 81a8cd7..6b4c9e1 100644
--- a/tests/test_generates.py
+++ b/tests/test_generates.py
@@ -1,6 +1,5 @@
 import sqlalchemy as sa
 from sqlalchemy_utils import generates
-from sqlalchemy_utils.decorators import generator
 from tests import TestCase