Fix return precedence in test