fixing viz bug