Update stage.py to track slightly modified crun API.