Add test for actual rewrite