Travis test fix