Rewrite test with co