Remove assert from tests (using should)