updating failing unit tests