diff --git a/context_windows/context.py b/context_windows/context.py
index 867458a..4f64f3f 100644
--- a/context_windows/context.py
+++ b/context_windows/context.py
@@ -1,6 +1,7 @@
 import numpy as np
 from typing import Callable, Optional, List
 import torch
+from ..utils import log
 
 def ordered_halving(val):
     bin_str = f"{val:064b}"