diff --git a/.github/workflows/train-dev.yml b/.github/workflows/train-dev.yml
index 2cc5a3f..0ce61c0 100644
--- a/.github/workflows/train-dev.yml
+++ b/.github/workflows/train-dev.yml
@@ -2,6 +2,12 @@ name: Train model in dev
 
 on:
   workflow_dispatch:
+  pull_request:
+    branches:
+      - main
+    paths:
+      - 'src/train-model-parameters.py'
+      - 'src/job.yml'
 
 permissions:
   contents: read
diff --git a/src/job.yml b/src/job.yml
index 9e6c769..9968624 100644
--- a/src/job.yml
+++ b/src/job.yml
@@ -7,8 +7,8 @@ command: >-
   --metrics_output ${{outputs.metrics_output}}
 inputs:
   training_data:
-    type: <uri_file_or_uri_folder>
-    path: <azureml:data-asset-name@latest>
+    type: uri_folder
+    path: azureml:diabetes-dev-folder@latest
   reg_rate: 0.01
 outputs:
   metrics_output:
diff --git a/src/train-model-parameters.py b/src/train-model-parameters.py
index 78581f5..2e6db8e 100644
--- a/src/train-model-parameters.py
+++ b/src/train-model-parameters.py
@@ -108,7 +108,7 @@ def parse_args():
     parser.add_argument("--training_data", dest='training_data',
                         type=str)
     parser.add_argument("--reg_rate", dest='reg_rate',
-                        type=float, default=0.01)
+                        type=float, default=0.00888)
     parser.add_argument("--metrics_output", dest='metrics_output',
                         type=str, default=None)