Hi Ben,
Just a quick, what is the type of your orbit? ( type(orbit) )
Sometimes one needs to cast types in the python version of orekit, this canin this case be done by:
orbit = KeplerianOrbit.cast_(orbit)
It is easier to troubleshoot if you post a cut-n-pastable test code / link to code.
Greetings,