rewrite test units