Fix and test broken category support
Merge pull request #8 from setowns1/bug

Fix and test broken category support