Also it's not necessary use `if` here, just return the expression