Added test for single-classification only module version