Wizard.py: add handling for async code