Update the tests to handle proper result type of (?:)