Aryna Sabalenka admits she should have taken a break from tennis after the death of her former boyfriend in March.