diff --git a/src/snowflake/sqlalchemy/functions.py b/src/snowflake/sqlalchemy/functions.py
index 4a7ee0b0..c08aa734 100644
--- a/src/snowflake/sqlalchemy/functions.py
+++ b/src/snowflake/sqlalchemy/functions.py
@@ -13,4 +13,4 @@ class flatten(sqlfunc.GenericFunction):
 
     def __init__(self, *args, **kwargs):
         warnings.warn(FLATTEN_WARNING, DeprecationWarning, stacklevel=2)
-        return super().__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)
diff --git a/tests/test_core.py b/tests/test_core.py
index 37f29832..c8a7775d 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -35,7 +35,7 @@
 )
 from sqlalchemy.exc import DBAPIError, NoSuchTableError
 from sqlalchemy.pool import NullPool
-from sqlalchemy.sql import and_, insert, not_, or_, select
+from sqlalchemy.sql import and_, not_, or_, select
 
 import snowflake.connector.errors
 import snowflake.sqlalchemy.snowdialect
@@ -1426,9 +1426,6 @@ def test_autoincrement(engine_testaccount):
         Column("name", String(39)),
     )
 
-    insert_stmt = insert(users)
-    select_stmt = select(users.c.name).order_by("uid")
-
     try:
         metadata.create_all(engine_testaccount)