Maybe a ternary operator here ?