diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index ec68b53..4ca7055 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -1,41 +1,45 @@
 # Name of the project in the GitHub action panel
 name: Run-Tests-and-Generate-Model-Diagram
 
-# Execute the CI on push on the master branch
+# Execute the CI on push or pull request on any branch
 on:
   push:
-    branches: 
-      - master
+    branches:
+      - '*'  # Allow the workflow to trigger on any branch
   pull_request:
     branches:
-      - master
+      - '*'  # Allow the workflow to trigger on any branch
   workflow_dispatch:
+
 jobs:
   build:
     runs-on: ubuntu-latest
     strategy:
       matrix:
         # List of images at https://github.com/hpi-swa/smalltalkCI#images
-        # Use Moose 10 that includes our visualization tool
+        # Use Moose64-11 that includes our visualization tool
         smalltalk: [ Moose64-11]
     name: ${{ matrix.smalltalk }}
     steps:
-        # checkout the project
-      - uses: actions/checkout@v4
+      # Checkout the code from the branch that triggered the workflow
+      - name: Checkout code
+        uses: actions/checkout@v4
+        with:
+          ref: ${{ github.ref }}  # Checkout the branch that triggered the workflow
       
-        # Prepare the CI - download the correct VM :-) 
+      # Prepare the CI - download the correct VM
       - uses: hpi-swa/setup-smalltalkCI@v1
         with:
           smalltalk-image: ${{ matrix.smalltalk }}
       
-        # Use the CI - always better to run test
+      # Run the tests
       - run: smalltalkci -s ${{ matrix.smalltalk }}
         shell: bash
         timeout-minutes: 15
         
       - name: Generate plantuml representation of meta-model
         run: |
-          $SMALLTALK_CI_VM  $SMALLTALK_CI_IMAGE eval "'FamixTypeScript.puml' asFileReference writeStreamDo: [ :stream | stream nextPutAll:  (FamixUMLDocumentor new model: FamixTypeScriptModel color: Color lightBlue ; beWithStubs ; generate; exportWith: (FamixUMLPlantUMLBackend new))]"
+          $SMALLTALK_CI_VM $SMALLTALK_CI_IMAGE eval "'FamixTypeScript.puml' asFileReference writeStreamDo: [ :stream | stream nextPutAll: (FamixUMLDocumentor new model: FamixTypeScriptModel color: Color lightBlue ; beWithStubs ; generate; exportWith: (FamixUMLPlantUMLBackend new))]"
    
       - name: Generate SVG Diagram from PlantUML
         uses: Timmy/plantuml-action@v1
@@ -57,4 +61,4 @@ jobs:
           git config --global user.email 'github-actions[bot]@users.noreply.github.com'
           git add doc-uml
           git commit -m "Update docs for branch ${{ github.ref_name }}"
-          git push origin ${{ github.ref_name }}  
+          git push origin HEAD:${{ github.ref_name }}  # Push to the branch that triggered the workflow