Fix to pass travis