diff --git a/tools/python_api/src_py/connection.py b/tools/python_api/src_py/connection.py
index 31da83ce0e..ceff10ad23 100644
--- a/tools/python_api/src_py/connection.py
+++ b/tools/python_api/src_py/connection.py
@@ -1,7 +1,6 @@
 from .query_result import QueryResult
 from .prepared_statement import PreparedStatement
 from . import _kuzu
-import numpy as np
 
 
 class Connection:
@@ -101,7 +100,7 @@ def _get_node_property_names(self, table_name):
             table_name)
         results = {}
         for (i, line) in enumerate(result_str.splitlines()):
-            #ignore first line
+            # ignore first line
             if i == 0:
                 continue
             line = line.strip()
@@ -181,4 +180,3 @@ def set_query_timeout(self, timeout_in_ms):
         """
 
         self._connection.set_query_timeout(timeout_in_ms)
-