diff --git a/src/main/java/org/example/petshopdesktop/controllers/InventoryController.java b/src/main/java/org/example/petshopdesktop/controllers/InventoryController.java
index d7cbcf97..f5f83461 100644
--- a/src/main/java/org/example/petshopdesktop/controllers/InventoryController.java
+++ b/src/main/java/org/example/petshopdesktop/controllers/InventoryController.java
@@ -232,7 +232,7 @@ public class InventoryController {
     private Inventory mapToInventory(InventoryResponse response) {
         return new Inventory(
             response.getInventoryId().intValue(),
-            0,
+            response.getProdId() != null ? response.getProdId().intValue() : 0,
             response.getProductName(),
             response.getCategoryName() != null ? response.getCategoryName() : "",
             0,