Aryna Sabalenka has revealed that it took a long time to recover from her shock loss to Madison Keys in the Australian Open final last month.