You could, but I've come around to the idea that you'd lose that battle.
There's no intervening event that separates the act of improving the lie.
So, two strokes is probably right.
Slugger was wrong though when he said "improving his line of play." It was improving his lie, not his line of play.