Aryna Sabalenka warned she broke tennis 'rule' that cost her Australian Open final