Fixing tests for travis.