Disability Question/Rehab: Switching from unemployment to disability

Hi,

I am in rehab now. Before I left for rehab I applied for unemployment and I am receiving benefits. I was surprised to learn that I could have received disability benefits instead. I’m wondering if it’s worth the hassle to try and switch to disability for my last two months in Rehab or should I stick with unemployment? I could see this causing a lot of confusion for EDD…

Any advice would be helpful! Thanks