Fix test clean up.