diff --git a/config/default.yaml b/config/default.yaml
index 7547712..ff78dee 100644
--- a/config/default.yaml
+++ b/config/default.yaml
@@ -1,5 +1,5 @@
 dataset_params:
-  im_path: 'data/train/images'
+  im_path: 'data/images'
 
 diffusion_params:
   num_timesteps : 1000