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

We should be expecting two attached sub classes in these unit tests.

Please review @caitp or @rillian.