Changed test logic for travis