Fixed code in accordance to pynbody update d2324e3