Use shouldReturn