Use better if statement

It looks better this way ...