diff --git a/yearn/outputs/victoria/output_helper.py b/yearn/outputs/victoria/output_helper.py
index 49e55ff87..aac1af91d 100644
--- a/yearn/outputs/victoria/output_helper.py
+++ b/yearn/outputs/victoria/output_helper.py
@@ -5,7 +5,6 @@
 import json
 from typing import List, Dict
 from brownie import chain
-from yearn.treasury.buckets import get_token_bucket
 from yearn.utils import contract
 from yearn.networks import Network
 
diff --git a/yearn/outputs/victoria/output_treasury.py b/yearn/outputs/victoria/output_treasury.py
index 78fb62978..e31dfd535 100644
--- a/yearn/outputs/victoria/output_treasury.py
+++ b/yearn/outputs/victoria/output_treasury.py
@@ -1,4 +1,5 @@
 from yearn.outputs.victoria.output_helper import _build_item, _post
+from yearn.treasury.buckets import get_token_bucket
 from yearn.utils import contract
 
 def export(timestamp, data, label):