Fixed bug variable name
Added a test
Version bump v1.0.4