fixing asserts for tests