1 files changed, 6 insertions, 6 deletions

diff --git a/beancount_extras_kris7t/plugins/default_tolerance_test.py b/beancount_extras_kris7t/plugins/default_tolerance_test.py
index 2e5c629..4ef84c1 100644
--- a/beancount_extras_kris7t/plugins/default_tolerance_test.py
+++ b/beancount_extras_kris7t/plugins/default_tolerance_test.py
@@ -3,13 +3,13 @@ __license__ = 'GNU GPLv2'
 
 import unittest
 
-from beancount import loader
+from beancount.loader import load_doc
 from beancount.parser import cmptest
 
 
 class DefaultToleranceTest(cmptest.TestCase):
 
-    @loader.load_doc()
+    @load_doc()
     def test_account_with_tolerance(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.default_tolerance"
@@ -28,7 +28,7 @@ class DefaultToleranceTest(cmptest.TestCase):
         2020-01-01 balance Assets:Checking  0 ~ 10 USD
         ''', entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_account_with_tolerance_override(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.default_tolerance"
@@ -47,7 +47,7 @@ class DefaultToleranceTest(cmptest.TestCase):
         2020-01-01 balance Assets:Checking  10 ~ 20 USD
         ''', entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_account_with_tolerance_override_zero(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.default_tolerance"
@@ -66,7 +66,7 @@ class DefaultToleranceTest(cmptest.TestCase):
         2020-01-01 balance Assets:Checking  0 ~ 0 USD
         ''', entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_account_without_tolerance(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.default_tolerance"
@@ -87,7 +87,7 @@ class DefaultToleranceTest(cmptest.TestCase):
         2020-01-02 balance Assets:Checking  10 ~ 20 USD
         ''', entries)
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_account_with_invalid_tolerance(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.default_tolerance"