diff --git a/mbed-tls-restricted-pr-test-parametrized.yaml b/mbed-tls-restricted-pr-test-parametrized.yaml
index 954377a..3525bb7 100644
--- a/mbed-tls-restricted-pr-test-parametrized.yaml
+++ b/mbed-tls-restricted-pr-test-parametrized.yaml
@@ -67,11 +67,13 @@
           Run TF-PSA-Crypto all.sh tests - Only enable if both TF_PSA_CRYPTO_REPO and TF_PSA_CRYPTO_BRANCH are set!
           Otherwise the test run will fail during BranchInfo gathering.
         name: RUN_TF_PSA_CRYPTO_ALL_SH
-    - string:
+    - git-parameter:
         name: TEST_BRANCH
         description: Branch or tag in https://github.com/Mbed-TLS/mbedtls-test.git (forks are not supported)
-        default: main
-        trim: true
+        type: PT_BRANCH
+        quickFilterEnabled: true
+        defaultValue: main
+        selectedValue: DEFAULT
     pipeline-scm:
       lightweight-checkout: false
       scm: