fix slight issue with column count