diff --git a/models/stock_location.py b/models/stock_location.py
index 91e8ba7..72cbaf0 100644
--- a/models/stock_location.py
+++ b/models/stock_location.py
@@ -28,8 +28,8 @@ class StockLocation(models.Model):
 
         # 3. Retrieve allowed locations from the picking type
         picking_type = self.env['stock.picking.type'].browse(picking_type_id)
-        if picking_type.exists() and picking_type.allowed_source_location_ids:
-            return picking_type.allowed_source_location_ids.ids
+        if picking_type.exists() and picking_type.default_location_src_ids:
+            return picking_type.default_location_src_ids.ids
             
         return []