diff --git a/.github/workflows/template_test.yml b/.github/workflows/template_test.yml
index 2ce9e02..8f63c1a 100644
--- a/.github/workflows/template_test.yml
+++ b/.github/workflows/template_test.yml
@@ -78,14 +78,14 @@ jobs:
           $testComposePath = "${{ inputs.testComposeFilePath }}"
 
           # create compose structure (using YAML structure for Docker Compose v2+)
-          $compose = @{ version = "3.8"; services = @{}; networks = @{} }
+          $compose = @{ services = @{}; networks = @{} }
           $compose.networks.test = @{ external = $false; name = "test-network-${{ github.run_id }}" } # Use unique network name per run
 
           # Generate services Section Based on Images inputs
           foreach ($img in $images) {
               $serviceName = $img.name
               $svc = @{}
-              $svc.image = "${{ env.IMAGE_REPO }}/$($serviceName):${{ env.TAG }}" # Use run_id tag
+              $svc.image = "${{ inputs.image_repo }}/$($serviceName):${{ env.TAG }}" # Use run_id tag
 
               if ($img.depends_on) {
                   $svc.depends_on = $img.depends_on
@@ -165,7 +165,7 @@ jobs:
               $ENV_ARGS \
               -v "$STAGING_DIR:$RESULTS_PATH" \
               --rm \
-              "${{ env.IMAGE_REPO }}/$($serviceName)/${{ inputs.testContainerName }}:${{ github.run_id }}" \
+              "${{ inputs.image_repo }}/$($serviceName)/${{ inputs.testContainerName }}:${{ github.run_id }}" \
               "$test_dir"
             # Add error handling if needed (e.g., exit script if a test run fails)
             if [ $? -ne 0 ]; then