Fixed the test for Feature/sorta