Unfortunately, I don't have a reference, as I was told this by an advisor from PWC a few years ago. I guess you could contact any good tax consultant, tell him the details, and let him look into it. I have no idea about how 401K works in the US.
https://uskanzlei.com/pages/usa-oecd-crs-facta-aia