Fixed missing heart beat state cleanup.