Extra HEAP_ALLOCED check.

Signed-off-by: Edward Z. Yang <ezyang@cs.stanford.edu>