Fix add category glitches
Merge pull request #346 from krazemon/add-category

Fix add category glitches