Changed to single line if statement