Fix broken feature code