Changing test to works with travis