Fix Travis test failure