[BUG] Test failing due to wrong sorting - #57