Update for travis-ci