Fix tests for .travis