1 files changed, 9 insertions, 9 deletions

diff --git a/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py b/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py
index 6ead45d..b4a9d8e 100644
--- a/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py
+++ b/beancount_extras_kris7t/plugins/selective_implicit_prices_test.py
@@ -6,15 +6,15 @@ import unittest
 
 from beancount.core.number import D
 from beancount.core import data
+from beancount.loader import load_doc
 from beancount.parser import cmptest
-from beancount import loader
 
 from beancount_extras_kris7t.plugins import selective_implicit_prices as implicit_prices
 
 
 class TestImplicitPrices(cmptest.TestCase):
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__all_cases(self, entries, _, options_map):
         """
         1702-04-02 commodity USD
@@ -124,7 +124,7 @@ class TestImplicitPrices(cmptest.TestCase):
             actual = (price.currency, price.amount.currency, price.amount.number)
             self.assertEqual(expected, actual)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__other_account(self, entries, errors, options_map):
         """
         2013-01-01 commodity HOOL
@@ -185,7 +185,7 @@ class TestImplicitPrices(cmptest.TestCase):
 
         """, new_entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__duplicates_on_same_transaction(self,
                                                                  entries, _, options_map):
         """
@@ -234,7 +234,7 @@ class TestImplicitPrices(cmptest.TestCase):
 
         """, new_entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__duplicates_on_different_transactions(self,
                                                                        entries, _,
                                                                        options_map):
@@ -296,7 +296,7 @@ class TestImplicitPrices(cmptest.TestCase):
 
         """, new_entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__duplicates_overloaded(self, entries, _, options_map):
         """
         2013-01-01 commodity HOOL
@@ -343,7 +343,7 @@ class TestImplicitPrices(cmptest.TestCase):
 
         """, new_entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__not_enabled(self, entries, errors, options_map):
         """
         2013-01-01 open Assets:Account1
@@ -363,7 +363,7 @@ class TestImplicitPrices(cmptest.TestCase):
           Assets:Other             -780000 USD
         """, new_entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__disabled(self, entries, errors, options_map):
         """
         2013-01-01 commodity HOOL
@@ -389,7 +389,7 @@ class TestImplicitPrices(cmptest.TestCase):
           Assets:Other             -780000 USD
         """, new_entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_add_implicit_prices__invalid(self, entries, errors, options_map):
         """
         2013-01-01 commodity HOOL