The exception allowed for disability is for total and permanent disability.
https://www.irs.gov/retirement-plans/plan-participant-employee/retirement-topics-tax-on-early-distri...
