diff --git a/main.py b/main.py
index cd19ee6ace0cf0659d76ce933c45da4852cceb14..61b0e12ef3bed25d081c545d20979a1e483fd018 100644
--- a/main.py
+++ b/main.py
@@ -25,7 +25,7 @@ def generate_folders(name, list_categories):
                 os.mkdir(f"{name}/{f}")
             else:
                 for cat in list_categories:
-                    is_exist2 = os.path.exists(f"{name}/Generated/{f}")
+                    is_exist2 = os.path.exists(f"{name}/Generated/{f}/{cat}")
                     if not is_exist2:
                         os.makedirs(f"{name}/Generated/{f}/{cat}")