Add header to the if statement

Missed it in prev pull req