Fixed a typo in vis.py