Fixed unit mismatches in Shaft Wizard