diff --git a/kotlin/android/app/src/main/java/dev/firezone/android/features/customuri/ui/CustomUriViewModel.kt b/kotlin/android/app/src/main/java/dev/firezone/android/features/customuri/ui/CustomUriViewModel.kt
index 42a58a6b8..3647af4f2 100644
--- a/kotlin/android/app/src/main/java/dev/firezone/android/features/customuri/ui/CustomUriViewModel.kt
+++ b/kotlin/android/app/src/main/java/dev/firezone/android/features/customuri/ui/CustomUriViewModel.kt
@@ -26,10 +26,10 @@ internal class CustomUriViewModel
         val actionLiveData: LiveData<ViewAction> = actionMutableLiveData
 
         fun parseCustomUri(intent: Intent) {
-            Log.d("CustomUriViewModel", "Parsing app link...")
+            Log.d("CustomUriViewModel", "Parsing callback...")
             viewModelScope.launch {
                 Log.d("CustomUriViewModel", "viewmodelScope.launch")
-                when (intent.data?.lastPathSegment) {
+                when (intent.data?.host) {
                     PATH_CALLBACK -> {
                         Log.d("CustomUriViewModel", "PATH_CALLBACK")
                         intent.data?.getQueryParameter(QUERY_CLIENT_STATE)?.let { state ->