Added assertions to tests