No one would make you take 9 months off if you lived in Canada, but you always have that option. You could also transfer some of your maternity leave to your partner if you were ready to go back to work earlier. Forward thinking countries recognize that flexibility is required for a healthy population and economy.