fix KeyError
Merge pull request #38 from yourilima/master

fix KeyError