I think if there’s one thing to be learned from previous missions, it’s that nothing will „just work“. If the ejection mechanism is damaged, or the spacecraft is hit by some object and rendered useless shortly before ejection, then the radioactive payload will make it to Earth. I think a roundtrip would be dangerous, and the core should be left at Mars.
Polluting the new homeland from the start, eh?
More seriously, we can afford to have a small amount of nuclear waste there, but we need to plan for it better than we have previously. First, document where it is and what it contains. Second, I don't know.
The correct response is reprocessing. Most of the "waste" generated by power reactors is actually still usable as fuel. It also eliminates the really long-lived stuff, so that the small amount of leftover waste only has to be kept stored for a much more manageable 500-1000 years.
Edit: even on a round trip it wouldn't be a big deal, the reactor can just be jettisoned before re-entering Earth's gravity well. You just can't have a fission reactor with spent fuel anywhere near Earth orbit. Space reactors in general are pretty interesting, because the design criteria are so different from terrestrial power reactors.