Fixing test for Travis CI