[Simulator.py] Fix imports.