A loaded walking calf raise will work the calf muscles whilst improving your balance.

Target an area in the gym where you are able to walk for 10-20m. Standing tall with a light DB in each hand, take small and slow steps forward walking in a straight line. With each step push off the ground with your toes to move up into a calf raise. Lower down, step forward and repeat on the opposite side.

Focus on propelling off the ground with each step but with control. This might test your balance!