Fix tests for travis