Fix multi class unit tests to expect the correct number of classes.
Merge pull request #195 from RickEyre/fix-multi-subclass

Fix multi class unit tests to expect the correct number of classes.