Add brackets around if statment