Change if statement to one liner just because.