Fix omission bug: Add `heap_segment' forward decl