FIX : viz test bug