Update tests for travis CI