fixing unit test problem