TST: clean up test