diff --git a/src/controller/warehouse.controller.js b/src/controller/warehouse.controller.js
index 555490b..968cd36 100644
--- a/src/controller/warehouse.controller.js
+++ b/src/controller/warehouse.controller.js
@@ -65,6 +65,10 @@ module.exports = {
 
     try {
       const warehouseDetails = await Warehouse.findById(id);
+      if (!warehouseDetails) {
+        res.send({ success: false, message: "ID not found" });
+        return;
+      }
       warehouseDetails.imageUrl = req.file.path;
       await warehouseDetails.save();
       res.send({ success: true, data: warehouseDetails });