diff --git a/exploration/ml_pipeline_classification.py b/exploration/ml_pipeline_classification.py
index 67514a3..d005785 100644
--- a/exploration/ml_pipeline_classification.py
+++ b/exploration/ml_pipeline_classification.py
@@ -52,14 +52,14 @@ SEGMENT_TYPE = "period"
 print("SEGMENT_TYPE: " + SEGMENT_TYPE)
 SEGMENT_LENGTH = "30_minutes_before"
 print("SEGMENT_LENGTH: " + SEGMENT_LENGTH)
-TARGET_VARIABLE = "appraisal_stressfulness"
+TARGET_VARIABLE = "JCQ_job_control"
 print("TARGET_VARIABLE: " + TARGET_VARIABLE)
 
-PATH_FULL = (
-    PATH_BASE
-    / SEGMENT_LENGTH
-    / ("input_" + TARGET_VARIABLE + "_" + SEGMENT_TYPE + "_mean.csv")
-)
+if "appraisal" in TARGET_VARIABLE:
+    TARGET_VARIABLE += "_"
+    TARGET_VARIABLE += SEGMENT_TYPE
+
+PATH_FULL = PATH_BASE / SEGMENT_LENGTH / ("input_" + TARGET_VARIABLE + "_mean.csv")
 
 model_input = pd.read_csv(PATH_FULL)