diff --git a/src/upload_handler.mjs b/src/upload_handler.mjs
index f5cdc41..ca3565b 100644
--- a/src/upload_handler.mjs
+++ b/src/upload_handler.mjs
@@ -155,6 +155,10 @@ export const handleUpload = async (req, res) => {
         const tmpPath = `./tmp/${filename}`;
         const destPath = `./public/b/${filename}`;
 
+        // Ensure directories exist
+        await fs.mkdir('./tmp', { recursive: true });
+        await fs.mkdir('./public/b', { recursive: true });
+
         // Save temporarily
         await fs.writeFile(tmpPath, file.data);