add test case and fix for issue #54
Merge pull request #59 from robocoder/patch-54

add test case and fix for issue #54