fix failing unit test
fixing errors