He wasn't forced to buy stock. It was a PR move so that he was injecting cash equivalent to Tesla's portion of the fine. However, he got stock in return, diluting the rest of the shareholders, so it was quite bizarre.

Foregoing reimbursement for private jet travel for a few years would have made actual sense, if his intent was to not punish Tesla for his actions.