diff --git a/ubc/calculator.py b/ubc/calculator.py
index f3ccc12..57e0ebf 100644
--- a/ubc/calculator.py
+++ b/ubc/calculator.py
@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from pandas import Timedelta
 
-from ubc.rates.openei.api import RateSchedule
+from ubc.rates.abstract import AbstractRate
 
 
 DELTA_HOUR = Timedelta("1 hour")
@@ -12,7 +12,7 @@ class SingleSite:
     """Single-Site Bill Calculator
     """
 
-    schedule: RateSchedule
+    schedule: AbstractRate
 
     def calculate_energy_charges(self, load):
         """
diff --git a/ubc/rates/openei/api.py b/ubc/rates/openei/api.py
index ac0c90b..910f2a6 100644
--- a/ubc/rates/openei/api.py
+++ b/ubc/rates/openei/api.py
@@ -4,8 +4,8 @@
 from dataclasses import dataclass, field
 from urllib.parse import urljoin
 
-from ubc.rates.abstract import AbstractRate
-from ubc.rates.openei.schemas import URDBMeta, Energy, Demand, FlatDemand, Meter
+from ..abstract import AbstractRate
+from .schemas import URDBMeta, Energy, Demand, FlatDemand, Meter
 
 
 root = "https://api.openei.org"