Fixing unit test for Travis-CI