fix to help demo_multiparty work properly.