Fix missing if brackets