If management believes that in-office work is more productive, that is their policy. You’re welcome to refuse, they’re welcome to fire you. They can change their mind as they wish, it is ‘at will’ employment. The company has the right to determine their policy.
That said, firing somebody because of a health issue is an outlier case that shouldn’t be used to justify the norm, it should be handled on an individual case basis.
Lawyers overreaching imho.
I predicted that they'd NEVER get back to "normal" if their work-from-home arrangement extended beyond the original "two weeks to flatten the curve."
I was right.