Fix remove_proxy tests