I agree with the riders that suggested breaks. Just before you think you’re going to get the behavior let your horse have a deserved break on a long rein. Then go back to collection.
Also remember that if you are trying to improve a gait, it is TRANSITIONS that improve the gait. Try lots of transitions to try and improve the length and quality of your collected work.
Good work and remember to keep it light.