fixing unit testing