fix draw.py bug