Minor improvement in simple sample test

What do you think about including this check?

I believe it keeps the test simple, but in addition it could useful for someone who is starting.