Fix bug in figuring out column.