spay onesie help

hello! i just got my cat spayed 2 days ago now and she was given back to us with a soft cone. she was okay with it on at first as she was still a little woozy, but once she wasn’t anymore all hell broke loose. she refused to keep it on, and if it wouldn’t come off, would pull on it constantly. i ordered her a spay onesie yesterday and tried it on her and she seemed to keep this one on, but her movements are limited. i know she can walk in it because if i call her over she low crouch walks to me then flops into my arms. but she walks and flops as she moves around or literally will roll to where she wants to be. i don’t want to go back to the cone as shes unpredictable with it on, will she eventually get used to the suit? anyone have a similar issue? id appreciate some advice. thank you!