diff --git a/web/src/flow/providers/oauth2/DeviceCodeFinish.ts b/web/src/flow/providers/oauth2/DeviceCodeFinish.ts
index 199b8aaaf..1813426eb 100644
--- a/web/src/flow/providers/oauth2/DeviceCodeFinish.ts
+++ b/web/src/flow/providers/oauth2/DeviceCodeFinish.ts
@@ -1,37 +1,25 @@
import "@goauthentik/elements/EmptyState";
import "@goauthentik/flow/FormStatic";
-import { AccessDeniedStage } from "@goauthentik/flow/stages/access_denied/AccessDeniedStage";
+import { BaseStage } from "@goauthentik/flow/stages/base";
import { t } from "@lingui/macro";
import { TemplateResult, html } from "lit";
import { customElement } from "lit/decorators.js";
+import { OAuthDeviceCodeFinishChallenge } from "@goauthentik/api";
+
@customElement("ak-flow-provider-oauth2-code-finish")
-export class DeviceCodeFinish extends AccessDeniedStage {
+export class DeviceCodeFinish extends BaseStage<
+ OAuthDeviceCodeFinishChallenge,
+ OAuthDeviceCodeFinishChallenge
+> {
render(): TemplateResult {
if (!this.challenge) {
return html`${this.challenge.flowInfo?.title}
-