Update trace.py

Looks like #1684 also ;)