Ever think of keyway broaching the flywheel and milling a slot on the
shaft. Maybe use 1/16" square key, or some of the key with a large
size on half for the flywheel and smaller for the shaft. You could
then solder or epoxy the key in place, and a slot rather than keyed
shaft would also prevent the key ever falling out.