diff --git a/openupgrade_scripts/scripts/stock/18.0.1.1/end-migration.py b/openupgrade_scripts/scripts/stock/18.0.1.1/end-migration.py
index 6f1449d6639..1694a729dda 100644
--- a/openupgrade_scripts/scripts/stock/18.0.1.1/end-migration.py
+++ b/openupgrade_scripts/scripts/stock/18.0.1.1/end-migration.py
@@ -4,13 +4,10 @@
 
 
 def fill_stock_picking_type_default_locations(env):
-    picking_types = env["stock.picking.type"].search(
-        [("default_location_src_id", "=", False)]
-    )
+    PickingType = env["stock.picking.type"].with_context(active_test=False)
+    picking_types = PickingType.search([("default_location_src_id", "=", False)])
     picking_types._compute_default_location_src_id()
-    picking_types = env["stock.picking.type"].search(
-        [("default_location_dest_id", "=", False)]
-    )
+    picking_types = PickingType.search([("default_location_dest_id", "=", False)])
     picking_types._compute_default_location_dest_id()