Fix the bug find_category!