Fixing unit test fail