diff --git a/python/datasketches/__init__.py b/python/datasketches/__init__.py
index 6b87303..a044e7f 100644
--- a/python/datasketches/__init__.py
+++ b/python/datasketches/__init__.py
@@ -1,4 +1,3 @@
-#from .streaming_heap import StreamingHeap
 from .count_sketch import CountSketch
 
 try:
diff --git a/python/datasketches/count_sketch.py b/python/datasketches/count_sketch.py
index 47cbf6a..3d3623e 100644
--- a/python/datasketches/count_sketch.py
+++ b/python/datasketches/count_sketch.py
@@ -17,7 +17,7 @@
 
 import numpy as np
 from random import randint, seed
-from datasketches.streaming_heap import StreamingHeap
+from .streaming_heap import StreamingHeap
 
 
 class CountSketch: