I am pretty sure that the payment is based on whether you were working before the baby was born, not whether you will return afterwards. Like Starfish said, check with Centrelink to be sure.