I suggest buying a spare wheel set from Lionel and turning them down on a lathe.You might end up with no way of holding the traction tires in place with out them the engine will not pull very good.
The front pilot axle being out of the pilot is very common to these engines. I have had three of the Flyer Chief versions go back for broken pilots (ONE TWICE) very poor design.
Gunny