Add tests for withHTML