diff --git a/delivery_driver/migrations/16.0.1.0.1/post-migration.py b/delivery_driver/migrations/16.0.1.0.1/post-migration.py
index eb644e20d2..218909fb11 100644
--- a/delivery_driver/migrations/16.0.1.0.1/post-migration.py
+++ b/delivery_driver/migrations/16.0.1.0.1/post-migration.py
@@ -12,7 +12,7 @@ def migrate(cr, version):
     with api.Environment.manage():
         env = api.Environment(cr, SUPERUSER_ID, {})
         partners = (
-            env["carrier.driver"]
+            env["delivery.carrier"]
             .search([("driver_id", "!=", False)])
             .mapped("driver_id")
         )