diff --git a/src/depiction/tools/cli/correct_baseline.py b/src/depiction/tools/cli/correct_baseline.py
index 6444a05..6f0033a 100644
--- a/src/depiction/tools/cli/correct_baseline.py
+++ b/src/depiction/tools/cli/correct_baseline.py
@@ -1,7 +1,8 @@
 from __future__ import annotations
 
 import shutil
-from typing import Annotated, Literal, TYPE_CHECKING
+from pathlib import Path
+from typing import Annotated, Literal
 
 import cyclopts
 import yaml
@@ -13,8 +14,6 @@
 from depiction.persistence import ImzmlReadFile, ImzmlWriteFile
 from depiction.tools.correct_baseline import BaselineVariants, CorrectBaseline
 
-from pathlib import Path
-
 app = cyclopts.App()
 
 
diff --git a/tests/unit/tools/simulate/test_generate_label_image.py b/tests/unit/tools/simulate/test_generate_label_image.py
index a0fb102..d9ee696 100644
--- a/tests/unit/tools/simulate/test_generate_label_image.py
+++ b/tests/unit/tools/simulate/test_generate_label_image.py
@@ -60,7 +60,7 @@ def test_render(generate) -> None:
     image = generate.render()
     assert image.n_channels == 3
     assert image.dimensions == (2, 2)
-    assert ["synthetic_0", "synthetic_1", "synthetic_2"] == image.channel_names
+    assert image.channel_names == ["synthetic_0", "synthetic_1", "synthetic_2"]
 
 
 if __name__ == "__main__":