Fixed the category picker