diff --git a/lantern/metric.py b/lantern/metric.py
index a382c55..9c01ae1 100644
--- a/lantern/metric.py
+++ b/lantern/metric.py
@@ -3,7 +3,8 @@
 
 import numpy as np
 
-from lantern import FunctionalBase, star
+from lantern.functional_base import FunctionalBase
+from lantern.star import star
 
 
 class MapMetric(FunctionalBase):