Small refactor to one line if statements