Fix heap warning