refactor: replace .map().flat() with .flatMap()

refactor: replace .map().flat() with .flatMap() #22129