It was a relatively quick fix that is likely "close" to a wash had they done it a different way.

But what you state is correct. I would have rather them granted a grace period to easily allow those companies to convert to a C corp in order to take advantage of the 21% corporate tax rate. After paying themselves a basic salary of $60K, they could then dividend any amount further than that to themselves.