Aryna Sabalenka fought through a tough challenge from Kamilla Rakhimova in her first match back from injury