Slight refactor
included if statement