Add more assertions in the tests.