diff --git a/tests/data/input/slice1.h5ad b/tests/data/input/slice1.h5ad
deleted file mode 100644
index e2dd51b..0000000
Binary files a/tests/data/input/slice1.h5ad and /dev/null differ
diff --git a/tests/data/input/slice2.h5ad b/tests/data/input/slice2.h5ad
deleted file mode 100644
index 2260734..0000000
Binary files a/tests/data/input/slice2.h5ad and /dev/null differ
diff --git a/tests/data/input/slice3.h5ad b/tests/data/input/slice3.h5ad
deleted file mode 100644
index 10fc00c..0000000
Binary files a/tests/data/input/slice3.h5ad and /dev/null differ
diff --git a/tests/data/input/slice4.h5ad b/tests/data/input/slice4.h5ad
deleted file mode 100644
index 782ae31..0000000
Binary files a/tests/data/input/slice4.h5ad and /dev/null differ
diff --git a/tests/test_align.py b/tests/test_align.py
index fd6f2aa..3b00ff4 100644
--- a/tests/test_align.py
+++ b/tests/test_align.py
@@ -1,6 +1,5 @@
 import pandas as pd
 import anndata as ad
-import pytest
 from pandas.testing import assert_frame_equal
 import scanpy as sc
 from pathlib import Path
@@ -16,25 +15,12 @@
 output_dir = test_dir / "data/output"
 
 
-@pytest.mark.parametrize(
-    "gene_fpath, spatial_fpath",
-    (
-        (
-            [f"{input_dir}/slice{i}.csv" for i in range(1, 4)],
-            [f"{input_dir}/slice{i}_coor.csv" for i in range(1, 4)],
-        ),
-        (
-            [f"{input_dir}/slice{i}.h5ad" for i in range(1, 4)],
-            None,
-        ),
-    ),
-)
-def test_cmd_line_center(tmp_path, gene_fpath, spatial_fpath):
+def test_cmd_line_center(tmp_path):
     print(f"Running command in {tmp_path}")
     result = align(
         "center",
-        gene_fpath,
-        spatial_fpath,
+        [f"{input_dir}/slice{i}.csv" for i in range(1, 4)],
+        [f"{input_dir}/slice{i}_coor.csv" for i in range(1, 4)],
         f"{tmp_path}",
         0.1,
         "kl",