Use if statement instead of return in shoulda test