Toddler bed

We just recently put my daughters toddler bed up (due with #2 in October and wanted to get her out of the crib) but she will not stay in it. She fights going to sleep anyways but if she falls asleep outside of her bed and we move her, she wakes up the second she hits the pillow. Or she will sleep maybe an hour and crawl out and cry. She has the same mattress, pillow, sheet, and cover as her crib just different frame. Any suggestions on how to help her (and me) sleep? I don’t want her thinking she can come in the bed with us after her brother is born.