Fix broken feature test