Bug fix - editing categories