just 1 category on test data