[BUG] Test failing due to wrong sorting #61