Fix test for Travis