Fix failing test on Travis-CI