move assertions to end of test