Could you change this to avoid having a one-liner condition?