diff --git a/configs/qa_short_ctx_self_gen.yaml b/configs/qa_short_ctx_self_gen.yaml
index d8608ae..5c8a6f0 100644
--- a/configs/qa_short_ctx_self_gen.yaml
+++ b/configs/qa_short_ctx_self_gen.yaml
@@ -39,11 +39,11 @@ target_modules:
 
 # data
 train_ds_names:
-  - self_gen/gemma-2-2b-it/fw_qa_v2_2k_len_level_3
-  - self_gen/gemma-2-2b-it/squad_compact
-  - self_gen/gemma-2-2b-it/pwc_compact
-  - self_gen/gemma-2-2b-it/ropes_compact
-  - self_gen/gemma-2-2b-it/drop_compact
+  - self_gen/google/gemma-2-2b-it/fw_qa_v2_2k_len_level_3
+  - self_gen/google/gemma-2-2b-it/squad_compact
+  - self_gen/google/gemma-2-2b-it/pwc_compact
+  - self_gen/google/gemma-2-2b-it/ropes_compact
+  - self_gen/google/gemma-2-2b-it/drop_compact
 
 val_ds_names:
   - squad
diff --git a/configs/qa_short_ctx_self_gen_no_fw_qa.yaml b/configs/qa_short_ctx_self_gen_no_fw_qa.yaml
index ecde052..de4165b 100644
--- a/configs/qa_short_ctx_self_gen_no_fw_qa.yaml
+++ b/configs/qa_short_ctx_self_gen_no_fw_qa.yaml
@@ -39,10 +39,10 @@ target_modules:
 
 # data
 train_ds_names:
-  - self_gen/gemma-2-2b-it/squad_compact
-  - self_gen/gemma-2-2b-it/pwc_compact
-  - self_gen/gemma-2-2b-it/ropes_compact
-  - self_gen/gemma-2-2b-it/drop_compact
+  - self_gen/google/gemma-2-2b-it/squad_compact
+  - self_gen/google/gemma-2-2b-it/pwc_compact
+  - self_gen/google/gemma-2-2b-it/ropes_compact
+  - self_gen/google/gemma-2-2b-it/drop_compact
 
 val_ds_names:
   - squad