Fixing tests in travis