add Fix bug in case there's only one column