Audi A4 2.0tdi oil pump

[AlfaSpider]

Going to look at a audi tonight it's a late 07 car with 130k miles. I'm told best to avoid with possible oil pump failure. Does this apply to the slightly later engines as in this car?
Any advice appreciated.

[Condi]

Fixed in 2010 I believe, and earlier ones have probably been done by now?

Its a shaft which rounds off, was replaced with a bigger/more substantial one in later models.