fix IfStatement