1 files changed, 3 insertions, 4 deletions
diff --git a/python/report_ops/cash.py b/python/report_ops/cash.py
index e1fd8319..84933c82 100644
--- a/python/report_ops/cash.py
+++ b/python/report_ops/cash.py
@@ -1,20 +1,19 @@
 import datetime
-import pandas as pd
 from typing import ClassVar
 from functools import partial
 from dataclasses import dataclass
+import pandas as pd
 
 from serenitas.utils.env import DAILY_DIR
 from serenitas.utils.db import dbconn
 from serenitas.analytics.dates import prev_business_day
 
-from .misc import get_dir, dt_from_fname, CUSTODIAN
-from .custodians import get_custodian_download_fun
+from .misc import get_dir, dt_from_fname, Custodian
 
 
 @dataclass
 class CashReport:
-    custodian: CUSTODIAN
+    custodian: Custodian
     date: datetime.date
     dtkey: ClassVar
     _conn: ClassVar[dbconn] = dbconn("dawndb")