diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 709fb964..0c2c9fc0 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -198,7 +198,7 @@ jobs:
             --resource_group_name ${{ env.AZURE_RESOURCE_GROUP_NAME }} \
             --workspace_name ${{ env.AZURE_ML_WORKSPACE_NAME }} \
             --location ${{ env.AZURE_LOCATION }} \
-            --tags "{\"triggering_actor\":\"${{github.triggering_actor}}\"}" \
+            --tags {\"git\":\"${{ github.head_ref }}.${{ github.sha }}\",\"version\":\"${{ needs.tags.outputs.new_version }}\",\"triggering_actor\":\"${{github.triggering_actor}}\"}" \
             > train_output.txt
           cat train_output.txt
       working-directory: train