Add tests for actual rewrite