Slight refactoring, put if statement on one line.