Fix Code for WarehouseMutation