diff --git a/tensorflow_datasets/image/nyu_depth_v2_test.py b/tensorflow_datasets/image/nyu_depth_v2_test.py
index <HASH>..<HASH> 100644
--- a/tensorflow_datasets/image/nyu_depth_v2_test.py
+++ b/tensorflow_datasets/image/nyu_depth_v2_test.py
@@ -10,7 +10,7 @@ from tensorflow_datasets.image import nyu_depth_v2
 
 class NyuDepthV2Test(testing.DatasetBuilderTestCase):
   DATASET_CLASS = nyu_depth_v2.NyuDepthV2
-  SPLITS = {"train": 2, "test": 1}
+  SPLITS = {"train": 2, "validation": 1}
 
 
 if __name__ == "__main__":
Fix validation split for NYU Depth V2 test