Small depth.py change