diff --git a/tests/tools/test_vi_model.py b/tests/tools/test_vi_model.py
index 241896d8..d8ecc0c8 100644
--- a/tests/tools/test_vi_model.py
+++ b/tests/tools/test_vi_model.py
@@ -1,11 +1,12 @@
-# isort: skip_file
-import pytest
 import contextlib
 
+import pytest
+
 import scvelo as scv
 
 with contextlib.suppress(ImportError):
     from scvi.data import synthetic_iid
+
     from scvelo.tools import VELOVI