Fix expression to get the right value to return