diff --git a/platforms/android/library/src/test/kotlin/io/element/android/wysiwyg/viewmodel/EditorViewModelTest.kt b/platforms/android/library/src/test/kotlin/io/element/android/wysiwyg/viewmodel/EditorViewModelTest.kt
index 95ae6e7a7..199a99e6f 100644
--- a/platforms/android/library/src/test/kotlin/io/element/android/wysiwyg/viewmodel/EditorViewModelTest.kt
+++ b/platforms/android/library/src/test/kotlin/io/element/android/wysiwyg/viewmodel/EditorViewModelTest.kt
@@ -214,12 +214,13 @@ internal class EditorViewModelTest {
 
     @Test
     fun `given internal edit link action, when get, it returns the right action`() {
-        composer.givenLinkAction(ComposerLinkAction.Edit(linkUrl))
+        composer.givenLinkAction(ComposerLinkAction.Edit(linkUrl, linkText))
 
         assertThat(
             viewModel.getLinkAction(), equalTo(
-                LinkAction.SetLink(
-                    currentUrl = linkUrl
+                LinkAction.EditLink(
+                    currentUrl = linkUrl,
+                    currentText = linkText
                 )
             )
         )