Fix bug in zprof.py