fixed failing unit tests
fixed failing unit test