fixed tests
Added new test for edge case