Fixes in visualization test.