Fix up transition.py includes