Fix missing header on error