Travis test fail fix