diff --git a/tests/implicitron/data/data_source.yaml b/tests/implicitron/data/data_source.yaml
index cb10134e..291c5504 100644
--- a/tests/implicitron/data/data_source.yaml
+++ b/tests/implicitron/data/data_source.yaml
@@ -90,8 +90,10 @@ dataset_map_provider_RenderedMeshDatasetMapProvider_args:
   num_views: 40
   data_file: null
   azimuth_range: 180.0
+  distance: 2.7
   resolution: 128
   use_point_light: true
+  gpu_idx: 0
   path_manager_factory_class_type: PathManagerFactory
   path_manager_factory_PathManagerFactory_args:
     silence_logs: true