Use __remove_mock_proxy per PR feedback