diff --git a/python/examples/test_dual_write_new.py b/python/examples/test_dual_write_new.py
index 751ae32..12d00a6 100644
--- a/python/examples/test_dual_write_new.py
+++ b/python/examples/test_dual_write_new.py
@@ -3,6 +3,8 @@
 
 #tag::test[]
 import unittest
+from pyspark.sql import SparkSession
+from pyspark.sql.functions import current_timestamp
 from pyspark.sql.types import Row
 from pyspark.testing.utils import assertDataFrameEqual
 from .dual_write import DualWriteExample
diff --git a/python/tox.ini b/python/tox.ini
index 330cd58..2aa2d4d 100644
--- a/python/tox.ini
+++ b/python/tox.ini
@@ -57,7 +57,7 @@ deps =
 [testenv:flake8]
 extras = tests
 skipsdist = True
-commands = flake8 --ignore=F403,E402,F401,F405,W503 examples
+commands = flake8 --ignore=F403,E402,F401,F405,W503,E265 examples
 allowlist_externals = flake8
 
 [testenv:mypy]