Fix failing proxy tests