Give him time to get used to using it again. Keep up the swim work.
If you are giving him attention when the leg is up try to stop doing that.
I had one dog that in her later years would still hold the 'poor little leg' up but would forget which one it was supposed to be - could all be a big play for attention!
Hope this helps you!
__________________
Please press Accept if you find my answer helpful. Want more info? Just reply on your question!
Bonuses for good work are always appreciated!
Thanks for the opportunity to help you!


