diff --git a/passbook/providers/oauth/views/oauth2.py b/passbook/providers/oauth/views/oauth2.py index 301d3e432..0196be52a 100644 --- a/passbook/providers/oauth/views/oauth2.py +++ b/passbook/providers/oauth/views/oauth2.py @@ -69,7 +69,7 @@ class AuthorizationFlowInitView(AccessMixin, View): PLAN_CONTEXT_NONCE: request.GET.get(PLAN_CONTEXT_NONCE), }, ) - plan.stages.append(in_memory_stage(OAuth2Stage)) + plan.append(in_memory_stage(OAuth2Stage)) self.request.session[SESSION_KEY_PLAN] = plan return redirect_with_qs( "passbook_flows:flow-executor-shell", diff --git a/passbook/providers/saml/views.py b/passbook/providers/saml/views.py index 8c182ecc3..9b2a3fb11 100644 --- a/passbook/providers/saml/views.py +++ b/passbook/providers/saml/views.py @@ -89,7 +89,7 @@ class SAMLSSOView(LoginRequiredMixin, SAMLAccessMixin, View): self.request, {PLAN_CONTEXT_SSO: True, PLAN_CONTEXT_APPLICATION: self.application}, ) - plan.stages.append(in_memory_stage(SAMLFlowFinalView)) + plan.append(in_memory_stage(SAMLFlowFinalView)) self.request.session[SESSION_KEY_PLAN] = plan return redirect_with_qs( "passbook_flows:flow-executor-shell",