diff --git a/tools/releasetools/sparse_img.py b/tools/releasetools/sparse_img.py
index 59d2b71..820d85b 100644
--- a/tools/releasetools/sparse_img.py
+++ b/tools/releasetools/sparse_img.py
@@ -207,9 +207,9 @@
     assert zero_blocks or nonzero_blocks
 
     if zero_blocks:
-      out["__ZERO"] = rangelib.RangeSet(data=zero_blocks)
+      out["__ZERO"] = RangeSet(data=zero_blocks)
     if nonzero_blocks:
-      out["__NONZERO"] = rangelib.RangeSet(data=nonzero_blocks)
+      out["__NONZERO"] = RangeSet(data=nonzero_blocks)
 
   def ResetFileMap(self):
     """Throw away the file map and treat the entire image as