Update for travis build