clean up unittest a little bit