I'm sorry you're having a problem with your Berkshire. This is EXACTLY why I didn't pre-order one of Lionel's new USRA heavy Mikados.
-First, the motor is undersized for the design and intended use. Premature failure was foreseeable.
-Second the worm gear and flywheel are permanently attached. So you don't need a motor. To get your loco running again, you need that EXACT motor with the flywheel and worm gear already attached. (The chances of pulling and pressing those parts onto another motor without bending a thin shaft like that are almost nil.)
The fact that K-Line took a permanent dirt nap not long after this loco was made doesn't help because the production volume was small. Maybe the corresponding part from a K-Line scale Mikado will work. If you do find a parts source I would buy two. That motor cost the manufacturer less than $5; with the flywheel and worm gear it should still cost you less than $50 "while supplies last."
Now compare this part to Lionel's recent reissue of the Lima Berkshire (parts diagram is on Lionel's website.) Lionel's Berk has a premium quality large motor with individually separable flywheel, dogbone shaft and worm gear. No problem--any good quality can motor will work. Of course the worm wheel is still captive in the chassis, but that's a gripe for a different day.
Maury Klein built them this way to reduce costs because he was trying to crack the scale market at a lower price point (~$600s instead of $800s.) I didn't bite then because quality is more important to me than saving money. Disappointment and frustration are always more expensive in the long run.
Lionel is taking the same shortcut now with their heavy Mikados. Partly to save money (by reusing the light Mikado chassis tooling), and partly to... accommodate whistle steam? A balky feature that you can barely see?? Give me a break (and a big motor in the firebox every day!) These are destined to be shelf queens. For about $100 they will make a nice static display on some executive's desk. My $.02.