Fix partner tests