Pietersen set to play in South Africa