diff --git a/src/controller/warehouse.controller.js b/src/controller/warehouse.controller.js
index 6aaae28..01c7a4a 100644
--- a/src/controller/warehouse.controller.js
+++ b/src/controller/warehouse.controller.js
@@ -60,8 +60,18 @@ module.exports = {
    */
   addWarehouseImage: async (req, res, next) => {
     // req.file contains the `warehouse-image`
-    console.dir({ file: req.file });
-    res.send("ok");
+    console.dir("Warehouse image uploaded:", { file: req.file });
+
+    const { id } = req.params;
+
+    try {
+      const warehouseDetails = await Warehouse.findById(id);
+      warehouseDetails.imageUrl = req.file.path;
+      await warehouseDetails.save();
+      res.send({ success: true, data: warehouseDetails });
+    } catch (err) {
+      next(err);
+    }
   },
 
   /**
diff --git a/src/controller/warehouse.router.js b/src/controller/warehouse.router.js
index 4c2e7cf..cd8bf99 100644
--- a/src/controller/warehouse.router.js
+++ b/src/controller/warehouse.router.js
@@ -16,7 +16,7 @@ router.post("/", controller.createWarehouse);
  * @route /warehouse/add-image
  */
 router.post(
-  "/add-image",
+  "/add-image/:id",
   upload.single("warehouse-image"),
   controller.addWarehouseImage
 );