The two engines are related, but not the same. The main change is to the cylinder head for obvious reasons - common rail instead of Pump Duse. The bottom end is very similar though, hence the carry over of the wretched oil pump/balancer module.
As others have said though the problem was licked after 2010, and I've only heard of a few isolated cases of early B8's with problems.