diff --git a/src/data/baseline_features.py b/src/data/baseline_features.py
index 61ea6eb7..918cba91 100644
--- a/src/data/baseline_features.py
+++ b/src/data/baseline_features.py
@@ -80,7 +80,7 @@ if not participant_info.empty:
                 .rename(columns={"index": "question", 0: "score_original"})
             )
             # Extract question IDs from names such as JobEisen[3]
-            limesurvey_demand.loc[:, "qid"] = (
+            limesurvey_demand["qid"] = (
                 limesurvey_demand["question"].str.extract(r"\[(\d+)\]").astype(int)
             )
             limesurvey_demand["score"] = limesurvey_demand["score_original"]
@@ -113,7 +113,7 @@ if not participant_info.empty:
                 .rename(columns={"index": "question", 0: "score_original"})
             )
             # Extract question IDs from names such as JobControle[3]
-            limesurvey_control.loc[:, "qid"] = (
+            limesurvey_control["qid"] = (
                 limesurvey_control["question"].str.extract(r"\[(\d+)\]").astype(int)
             )
             limesurvey_control["score"] = limesurvey_control["score_original"]