diff --git a/prepare_data.py b/prepare_data.py
index 7bcad7addc30a7d8ca157af0579c162ea57cac59..4076978d080c56331d4184c958497ef4762639a5 100644
--- a/prepare_data.py
+++ b/prepare_data.py
@@ -362,7 +362,8 @@ if __name__ == '__main__':
         target = 'yield'
 
         wd_dataset_airbnb = prepare_data_airbnb(
-            df = DF.sample(5000),
+            # df = DF.sample(5000),
+            df = DF.sample,
             img_id = img_id,
             img_path = img_path,
             text_col = text_col,