diff --git a/src/matrix/registration/Registration.ts b/src/matrix/registration/Registration.ts
index c9c9af87..0d342639 100644
--- a/src/matrix/registration/Registration.ts
+++ b/src/matrix/registration/Registration.ts
@@ -18,6 +18,7 @@ import type {HomeServerApi} from "../net/HomeServerApi";
 import type {BaseRegistrationStage} from "./stages/BaseRegistrationStage";
 import {DummyAuth} from "./stages/DummyAuth";
 import {TermsAuth} from "./stages/TermsAuth";
+import {TokenAuth, TOKEN_AUTH_TYPE} from "./stages/TokenAuth";
 import type {
     AccountDetails,
     RegistrationFlow,
@@ -108,6 +109,8 @@ export class Registration {
                 return new DummyAuth(session, params?.[type]);
             case "m.login.terms":
                 return new TermsAuth(session, params?.[type]);
+            case TOKEN_AUTH_TYPE:
+                return new TokenAuth(session, params?.[type]);
             default:
                 throw new Error(`Unknown stage: ${type}`);
         }