Fixes error when state column does not exists
Merge pull request #2 from jrichardlai/fixes-when-state-column-does-not-exist

Fixes error when state column does not exists