fix imports in plan.py