diff --git a/tests/test_sale_shipment_cost.py b/tests/test_sale_shipment_cost.py
index 5a19c6934e..0f987b3cf1 100644
--- a/tests/test_sale_shipment_cost.py
+++ b/tests/test_sale_shipment_cost.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 SaleShipmentCostTestCase(unittest.TestCase):
@@ -26,6 +27,6 @@ def suite():
     suite.addTests(unittest.TestLoader().loadTestsFromTestCase(
         SaleShipmentCostTestCase))
     suite.addTests(doctest.DocFileSuite('scenario_sale_shipment_cost.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