1 files changed, 2 insertions, 5 deletions
diff --git a/python/collateral/citi.py b/python/collateral/citi.py
index acda4b1f..bf3cee16 100644
--- a/python/collateral/citi.py
+++ b/python/collateral/citi.py
@@ -1,6 +1,6 @@
 import pandas as pd
 from pandas.tseries.offsets import BDay
-from . import DAILY_DIR, bus_day
+from . import DAILY_DIR
 from .common import load_pdf, get_col
 
 
@@ -16,10 +16,7 @@ def load_file(d):
     return pd.read_excel(fname, skiprows=6, skipfooter=2)
 
 
-def download_files(count=20):
-    from exchange import ExchangeMessage
-
-    em = ExchangeMessage()
+def download_files(em, count=20):
     emails = em.get_msgs(
         path=["NYops", "Margin Calls Citi"], count=count, subject__startswith="262966"
     )