diff --git a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/auth/registration/RegistrationWizard.kt b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/auth/registration/RegistrationWizard.kt
index 995fd27acecfd2241b099b4f9c5e4e502feb2ddf..85600e279f128d336c77c39ae8bc499d4621a340 100644
--- a/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/auth/registration/RegistrationWizard.kt
+++ b/matrix-sdk-android/src/main/java/org/matrix/android/sdk/api/auth/registration/RegistrationWizard.kt
@@ -82,7 +82,7 @@ interface RegistrationWizard {
      * Current registration "session" param will be included into authParams by default.
      * The authParams should contain at least one entry "type" with a String value.
      */
-    suspend fun registrationCustom(authParams: JsonDict): RegistrationResult
+    suspend fun registrationCustom(authParams: JsonDict, initialDeviceDisplayName: String? = null): RegistrationResult
 
     /**
      * Perform the "m.login.email.identity" or "m.login.msisdn" stage.
@@ -121,4 +121,7 @@ interface RegistrationWizard {
      * called successfully.
      */
     fun isRegistrationStarted(): Boolean
-}
+
+    //Added to support few registration flows
+    suspend fun getAllRegistrationFlows(): List<List<Stage>>
+}
\ No newline at end of file