Minimal changes in TableVisualization for testing