Update for travis ci