Sabalenka appeared on course for victory after surging ahead in the final set, before allowing Rybakina back into the match.