Updated test for additional total column