Update for travis CI