fix learners unit test
clean up