diff --git a/test/external/external_test_datasets.py b/test/external/external_test_datasets.py
index bf1b4d30c3..0cd0fceaa8 100644
--- a/test/external/external_test_datasets.py
+++ b/test/external/external_test_datasets.py
@@ -28,7 +28,7 @@ class TestKiTS19Dataset(ExternalTestDatasets):
   def _create_samples(self, val, num_samples=2):
     self._set_seed()
 
-    img, lbl = np.random.rand(95, 196, 196).astype(np.float32), np.random.randint(0, 100, size=(95, 196, 196)).astype(np.uint8)
+    img, lbl = np.random.rand(190, 392, 392).astype(np.float32), np.random.randint(0, 100, size=(190, 392, 392)).astype(np.uint8)
     img, lbl = nib.Nifti1Image(img, np.eye(4)), nib.Nifti1Image(lbl, np.eye(4))
     dataset = "val" if val else "train"
     preproc_pth = Path(tempfile.gettempdir() + f"/{dataset}")