diff --git a/tests/test_init.py b/tests/test_init.py
index 0cb567b..3b90931 100644
--- a/tests/test_init.py
+++ b/tests/test_init.py
@@ -7,18 +7,10 @@
 
 @pytest.mark.parametrize("optional_dep", ["pandas", "numpy", "pyarrow"])
 def test_optional_dependencies(optional_dep):
-    with mock.patch.dict(sys.modules, {'optional_dependency': None}):
+    with mock.patch.dict(sys.modules, {"optional_dependency": None}):
         if "ibis_ml" in sys.modules:
             reload(sys.modules["ibis_ml"])
         else:
             import_module("ibis_ml")
 
         assert "ibis_ml" in sys.modules
-
-
-def test_import():
-    try:
-        import_module("ibis_ml")
-    except AttributeError:
-        raise ImportError("cannot import IbisML") from None
-    assert "ibis_ml" in sys.modules