Rework awkward if statement