diff --git a/tests/_ground_truth_figures/plot_tsi.png b/tests/_ground_truth_figures/plot_tsi.png
new file mode 100644
index 000000000..d84e18a5a
Binary files /dev/null and b/tests/_ground_truth_figures/plot_tsi.png differ
diff --git a/tests/test_plotting.py b/tests/test_plotting.py
index ec724eac5..134b46e9b 100644
--- a/tests/test_plotting.py
+++ b/tests/test_plotting.py
@@ -2257,6 +2257,13 @@ def test_scvelo_transition_matrix_projection(self, mc: GPCCA, fpath: str):
             save=fpath,
         )
 
+    @compare(kind="gpcca")
+    def test_plot_tsi(self, mc: GPCCA, fpath: str):
+        terminal_states = ["Neuroblast", "Astrocyte", "Granule mature"]
+        cluster_key = "clusters"
+        _ = mc.tsi(n_macrostates=3, terminal_states=terminal_states, cluster_key=cluster_key, n_cells=10)
+        mc.plot_tsi(dpi=DPI, save=fpath)
+
 
 class TestLineage:
     @compare(kind="lineage")