diff --git a/include/aeb_heap.hrl b/include/aeb_heap.hrl
index a37d97a..ba8c977 100644
--- a/include/aeb_heap.hrl
+++ b/include/aeb_heap.hrl
@@ -3,13 +3,13 @@
                val_t  :: aeso_sophia:type(),
                parent :: none | non_neg_integer(),
                size   = 0 :: non_neg_integer(),
-               data   :: #{aeso_heap:binary_value() => aeso_heap:binary_value() | tombstone}
+               data   :: #{aeb_heap:binary_value() => aeb_heap:binary_value() | tombstone}
                        | stored}).
 
 -record(maps, { maps    = #{} :: #{ non_neg_integer() => #pmap{} }
               , next_id = 0   :: non_neg_integer() }).
 
 -record(heap, { maps   :: #maps{},
-                offset :: aeso_heap:offset(),
+                offset :: aeb_heap:offset(),
                 heap   :: binary() | #{non_neg_integer() => non_neg_integer()} }).