Edit: fix_nonzero.py