Updated assertions to modified tests.