18v Milwaukee Fuel pops or slips out of gear

Replacing the gear train assembly on my Milwaukee 18 volt Fuel drill