Small change to vis script