Inferred Set[Any]. Add explicit if this is correct

Lint issue