diff --git a/addons/stock_picking_batch/models/__init__.py b/addons/stock_picking_batch/models/__init__.py
index 6f4ec3d8036..a5f3f55679c 100644
--- a/addons/stock_picking_batch/models/__init__.py
+++ b/addons/stock_picking_batch/models/__init__.py
@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
 
+from . import stock_picking
 from . import stock_picking_batch
diff --git a/addons/stock_picking_batch/models/stock_picking.py b/addons/stock_picking_batch/models/stock_picking.py
new file mode 100644
index 00000000000..b7501f41de0
--- /dev/null
+++ b/addons/stock_picking_batch/models/stock_picking.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models
+
+
+class StockPicking(models.Model):
+    _inherit = "stock.picking"
+
+    batch_id = fields.Many2one(
+        'stock.picking.batch', string='Batch Transfer',
+        check_company=True,
+        states={'done': [('readonly', True)], 'cancel': [('readonly', True)]},
+        help='Batch associated to this transfer', copy=False)
+
+    def _should_show_transfers(self):
+        if len(self.batch_id) == 1 and self == self.batch_id.picking_ids:
+            return False
+        return super()._should_show_transfers()
diff --git a/addons/stock_picking_batch/models/stock_picking_batch.py b/addons/stock_picking_batch/models/stock_picking_batch.py
index eda23f1b60a..70e4c1b17f2 100644
--- a/addons/stock_picking_batch/models/stock_picking_batch.py
+++ b/addons/stock_picking_batch/models/stock_picking_batch.py
@@ -75,18 +75,3 @@ class StockPickingBatch(models.Model):
             return self.env.ref('stock_picking_batch.mt_batch_state')
         return super(StockPickingBatch, self)._track_subtype(init_values)
 
-
-class StockPicking(models.Model):
-    _inherit = "stock.picking"
-
-    batch_id = fields.Many2one(
-        'stock.picking.batch', string='Batch Transfer',
-        check_company=True,
-        states={'done': [('readonly', True)], 'cancel': [('readonly', True)]},
-        help='Batch associated to this transfer', copy=False)
-
-    def _should_show_transfers(self):
-        if len(self.batch_id) == 1 and self == self.batch_id.picking_ids:
-            return False
-        return super()._should_show_transfers()
-