Improved test coverage for proxy feature