fix broken feature