diff --git a/auth/app/src/main/java/com/google/firebase/quickstart/auth/MainActivity.java b/auth/app/src/main/java/com/google/firebase/quickstart/auth/MainActivity.java
index 88fb92092..5b6aa1652 100644
--- a/auth/app/src/main/java/com/google/firebase/quickstart/auth/MainActivity.java
+++ b/auth/app/src/main/java/com/google/firebase/quickstart/auth/MainActivity.java
@@ -340,6 +340,7 @@ public void buildActionCodeSettings() {
                         .setUrl("https://www.example.com/finishSignUp?cartId=1234")
                         // This must be true
                         .setHandleCodeInApp(true)
+                        .setLinkDomain("custom-domain.com")
                         .setIOSBundleId("com.example.ios")
                         .setAndroidPackageName(
                                 "com.example.android",
diff --git a/auth/app/src/main/java/com/google/firebase/quickstart/auth/kotlin/MainActivity.kt b/auth/app/src/main/java/com/google/firebase/quickstart/auth/kotlin/MainActivity.kt
index 17d3bc17e..dc0587864 100644
--- a/auth/app/src/main/java/com/google/firebase/quickstart/auth/kotlin/MainActivity.kt
+++ b/auth/app/src/main/java/com/google/firebase/quickstart/auth/kotlin/MainActivity.kt
@@ -280,6 +280,7 @@ abstract class MainActivity : AppCompatActivity() {
             url = "https://www.example.com/finishSignUp?cartId=1234"
             // This must be true
             handleCodeInApp = true
+            setLinkDomain("custom-domain.com")
             setIOSBundleId("com.example.ios")
             setAndroidPackageName(
                 "com.example.android",