Merge pull request #227 from geek/master

Fixing test
Adding test for missing coverage