Category not found fix