@jennyr If you were walking with an altered gait prior to surgery it will take longer to get back to normal. Be sure to consciously try to have a normal gait so you don’t just fall into a habit of walking awkwardly. Hydrotherapy is great for rehab, so keep it up. I was limping terribly before my first TKR so it did take longer to get back to normal than it did after my second TKR.
Keep working at it, it will happen.