Changed test to work with github travis ci