When you are working to improve and area of your swing, don’t get so caught up with ball flight. If you use the ball flight solely as your basis for improving your swing, you will not be improving your swing. You will continue too adjust your swing to make the ball go straight.
In other words, continue to work on making the swing correction regardless of ball flight. Once the correction is ingrained into your swing, then the ball flight will change itself.