diff --git a/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py b/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
index 143d5b78bd51..0eae437eb0f8 100644
--- a/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
+++ b/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
@@ -53,7 +53,7 @@ def test_init_with_non_string_drop_columns(self) -> None:
         with self.assertRaises(ValueError):
             VerticalEvenPartitioner(num_partitions=2, drop_columns=[1, "a", 3])
 
-        def test_init_with_non_list_drop_columns(self) -> None:
+    def test_init_with_non_list_drop_columns(self) -> None:
         """Test initialization with non-list elements in drop_columns."""
         with self.assertRaises(ValueError):
             VerticalEvenPartitioner(num_partitions=2, drop_columns="a")