added assertions to test