Left-arm wristspinner is returning to India to prepare for the upcoming Test series against South Africa