diff --git a/tests/luna/pathology/cli/test_extract_shape_features.py b/tests/luna/pathology/cli/test_extract_shape_features.py
index 3b4b8335..bc0177cb 100644
--- a/tests/luna/pathology/cli/test_extract_shape_features.py
+++ b/tests/luna/pathology/cli/test_extract_shape_features.py
@@ -23,4 +23,4 @@ def test_cli_generate_mask(tmp_path):
     assert os.path.exists(f"{tmp_path}/metadata.yml")
     df = pd.read_csv(f"{tmp_path}/shape_features.csv")
 
-    assert len(df) == 2
+    assert len(df) == 178