Updated tests for travis