Category edit/new fixed