Sabalenka getting in the mood for clay in Stuttgart