minor change, put brackets around if statement