diff --git a/openfl-workspace/experimental/101_torch_cnn_mnist/plan/data.yaml b/openfl-workspace/experimental/101_torch_cnn_mnist/plan/data.yaml
index 1f051ef765c..5cda1a7e9d8 100644
--- a/openfl-workspace/experimental/101_torch_cnn_mnist/plan/data.yaml
+++ b/openfl-workspace/experimental/101_torch_cnn_mnist/plan/data.yaml
@@ -20,8 +20,8 @@ col2:
   callable_func:
     settings:
       batch_size: 32
-      index: 1
+      index: 2
       n_collaborators: 2
       test_dataset: src.collaborator_private_attrs.train_dataset
       train_dataset: src.collaborator_private_attrs.test_dataset
-    template: src.collaborator_private_attrs.collaborator_private_attrs
\ No newline at end of file
+    template: src.collaborator_private_attrs.collaborator_private_attrs