Fixing travis ci tests