Fix failing test on Travis CI