diff --git a/tests/e2e_test/test_e2e.py b/tests/e2e_test/test_e2e.py
index b66014aa..b37aa8f5 100644
--- a/tests/e2e_test/test_e2e.py
+++ b/tests/e2e_test/test_e2e.py
@@ -11,7 +11,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import math
 import subprocess
 import asyncio
 import pytest
diff --git a/tests/e2e_test/test_migration.py b/tests/e2e_test/test_migration.py
index 028a5f29..ed574afc 100644
--- a/tests/e2e_test/test_migration.py
+++ b/tests/e2e_test/test_migration.py
@@ -11,7 +11,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import math
 import asyncio
 from collections import defaultdict
 import re