diff --git a/tests/test_sale_shipment_grouping.py b/tests/test_sale_shipment_grouping.py
index 69a06a417e..df4b4da94e 100644
--- a/tests/test_sale_shipment_grouping.py
+++ b/tests/test_sale_shipment_grouping.py
@@ -3,7 +3,8 @@
 import unittest
 import doctest
 import trytond.tests.test_tryton
-from trytond.tests.test_tryton import test_view, test_depends, doctest_dropdb
+from trytond.tests.test_tryton import test_view, test_depends
+from trytond.tests.test_tryton import doctest_setup, doctest_teardown
 
 
 class SaleShipmentGroupingTestCase(unittest.TestCase):
@@ -26,6 +27,6 @@ def suite():
     loader = unittest.TestLoader()
     suite.addTests(loader.loadTestsFromTestCase(SaleShipmentGroupingTestCase))
     suite.addTests(doctest.DocFileSuite('scenario_sale_shipment_grouping.rst',
-            setUp=doctest_dropdb, tearDown=doctest_dropdb, encoding='utf-8',
+            setUp=doctest_setup, tearDown=doctest_teardown, encoding='utf-8',
             optionflags=doctest.REPORT_ONLY_FIRST_FAILURE))
     return suite