diff --git a/.github/workflows/code_quality.yml b/.github/workflows/code_quality.yml
index 635687c..b3be658 100644
--- a/.github/workflows/code_quality.yml
+++ b/.github/workflows/code_quality.yml
@@ -13,6 +13,9 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v3
+      with:
+        persist-credentials: false  # Important: This must be set for actions-js/push
+        fetch-depth: 0               # Important: Fetch all history for all tags and branches
     - name: Set up Python
       uses: actions/setup-python@v4
       with:
@@ -25,8 +28,8 @@ jobs:
       run: ruff --fix .
     - name: Security Check with Bandit
       run: bandit -r .
-    # Commit and push changes made by Black and Ruff
     - name: Commit & Push changes
-      uses: actions-js/push@main
+      uses: actions-js/push@v1.4
       with:
         github_token: ${{ secrets.MY_GITHUB_TOKEN }}
+        branch: ${{ github.ref }}
\ No newline at end of file
diff --git a/src/energy_forcasting_model/pipelines/lightgbm_training_pipeline/pipeline.py b/src/energy_forcasting_model/pipelines/lightgbm_training_pipeline/pipeline.py
index 0c28a59..dfa9d58 100644
--- a/src/energy_forcasting_model/pipelines/lightgbm_training_pipeline/pipeline.py
+++ b/src/energy_forcasting_model/pipelines/lightgbm_training_pipeline/pipeline.py
@@ -51,10 +51,10 @@ def create_pipeline(**kwargs) -> Pipeline:
         tags="model_training",
         namespace="lightgbm_training_pipeline",
         inputs=[
-            "X_train",
-            "y_train",
-            "X_test",
-            "y_test",
+                        "X_train",
+                        "y_train",
+                        "X_test",
+                        "y_test",
         ],
         outputs=[
 "lightgbm_model",