Updated number of columns in unit test