diff --git a/assets/js/butter-dir-listing.js b/assets/js/butter-dir-listing.js
index ad5d545..44f9b12 100644
--- a/assets/js/butter-dir-listing.js
+++ b/assets/js/butter-dir-listing.js
@@ -145,7 +145,9 @@ window.onload = function () {
     // Create a folder listing
     const folderListing = document.createElement("div");
     folderListing.classList.add('folder-list');
-    folderListing.innerHTML = '<h2>Folders</h2>';
+    if (folders.length > 0) {
+        folderListing.innerHTML = '<h2>Folders</h2>';
+    }
 
     // for each folder, add a div to folderListing
     folders.forEach(folder => {
@@ -165,7 +167,9 @@ window.onload = function () {
     // then for files
     const fileListing = document.createElement("div");
     fileListing.classList.add('file-list');
-    fileListing.innerHTML = '<h2>Files</h2>';
+    if (files.length > 0) {
+        fileListing.innerHTML = '<h2>Files</h2>';
+    }
 
     // for each file, add a div to fileListing
     files.forEach(file => {