Quick fix to replace the authentication probability with the key probability