Shaft alignment optimization