fix index error in shallow column
simplified handling index, add more test
Merge pull request #361 from kmuto/fix-shallow-column

fix index error in shallow column