Fixes #1, actually the fix for issue #11 also fixed this issue, but the test case was not correctly. Here we fix the test case and the expected results