Fix test failure on Travis