add asserts and test data which seemed to have been removed somehow