update wasp.py