removing asserts and making the tests expect like