diff --git a/urls/main.py b/urls/main.py
index cfed93f..12c4e06 100644
--- a/urls/main.py
+++ b/urls/main.py
@@ -40,7 +40,7 @@ def save_urls(urls, output_file, root_folder):
                 for file in files:
                     f.write(f"  {file}\n")
             else:
-                f.write(f"  {dir_name}:\n")
+                f.write(f"  {dir_name.split('/')[-1]}:\n")
                 for file in files:
                     f.write(f"    {file}\n")