Kolisi to return for Springboks if he proves his fitness