Fixing unit test for correct sorting.