Update tests for travis