Fix failing test on Travis