this works for the test sets