Product review (4 stars): Replacement with modifications
As other reviewers advised, I drilled the drive shaft and used a cotter pin to hold the plastic cup in place (see picture).  I used a Traxxas body clip sold on Amazon for the cotter pin.