bug fix - need to make sure the column size is correct