For your case, maybe you can ask a semester leave from your school, and just go back to Canada about three months before the baby's due date, and come back to US right after. It will be a hard and long trips for you, but avoid the citizenship hassle.