Fix visualization test.