The correct part is the stub shaft from the 1.6L, which is an oem part. There are however non-oem stub shafts available, whick lack the oil groove.

Jacks isn't the only one who changed his mind on balance shafts, AMS, while a little different suggests using their "race" balance shaft on the oil pump.