diff --git a/.github/workflows/openfl-test.yml b/.github/workflows/openfl-test.yml
index 86ce00a4d..4c48b54b7 100644
--- a/.github/workflows/openfl-test.yml
+++ b/.github/workflows/openfl-test.yml
@@ -90,15 +90,17 @@ jobs:
         git clone --depth=1 https://github.com/securefederatedai/openfl.git
         cd openfl
         git fetch --tags
-        echo "Checkout the latest OpenFL tag"
-        latestTag=$(git describe --tags "$(git rev-list --tags --max-count=1)")
-        git checkout $latestTag
-        sed -i -e 's/protobuf==3.19.6/protobuf/g' setup.py ## this should NOT be there
+        # echo "Checkout the latest OpenFL tag"
+        # latestTag=$(git describe --tags "$(git rev-list --tags --max-count=1)")
+        # git checkout $latestTag
+        # sed -i -e 's/protobuf==3.19.6/protobuf/g' setup.py ## this should NOT be there
         pip install -e .
         cd ..
         echo "Copying files to appropriate directories and updated headers"
-        head -n 1 testing/data/train_3d_rad_segmentation.csv > /home/runner/work/GaNDLF/GaNDLF/openfl/valid.csv
-        tail -n +9 testing/data/train_3d_rad_segmentation.csv >> /home/runner/work/GaNDLF/GaNDLF/openfl/valid.csv
-        head -n 8 testing/data/train_3d_rad_segmentation.csv > /home/runner/work/GaNDLF/GaNDLF/openfl/train.csv
+        head -n 1 testing/data/train_2d_rad_segmentation.csv > /home/runner/work/GaNDLF/GaNDLF/openfl/valid.csv
+        tail -n +9 testing/data/train_2d_rad_segmentation.csv >> /home/runner/work/GaNDLF/GaNDLF/openfl/valid.csv
+        head -n 8 testing/data/train_2d_rad_segmentation.csv > /home/runner/work/GaNDLF/GaNDLF/openfl/train.csv
+        sed -i 's/# n_channels: 3/num_channels: 3/g' testing/config_segmentation.yaml
+        config_to_use=$(pwd)/testing/config_segmentation.yaml
         cd openfl
-        python -m tests.github.test_gandlf --template gandlf_seg_test --fed_workspace aggregator --col1 one --col2 two --rounds-to-train 1
+        python -m tests.github.test_gandlf --template gandlf_seg_test --fed_workspace aggregator --col1 one --col2 two --rounds-to-train 1 --gandlf_config $config_to_use