Make logic be a one-liner to make rubocop happy