Maybe a ternary operator here ? (also)