diff --git a/vaadin-button-flow-parent/vaadin-button-flow/pom.xml b/vaadin-button-flow-parent/vaadin-button-flow/pom.xml
index 0d4fa5e6ba0..5b55b93b8b1 100644
--- a/vaadin-button-flow-parent/vaadin-button-flow/pom.xml
+++ b/vaadin-button-flow-parent/vaadin-button-flow/pom.xml
@@ -56,6 +56,11 @@
jakarta.jakartaee-web-api
test
+
+ org.mockito
+ mockito-core
+ test
+
diff --git a/vaadin-button-flow-parent/vaadin-button-flow/src/main/java/com/vaadin/flow/component/button/Button.java b/vaadin-button-flow-parent/vaadin-button-flow/src/main/java/com/vaadin/flow/component/button/Button.java
index 9dbfee9d053..1a2c9d8ae4c 100644
--- a/vaadin-button-flow-parent/vaadin-button-flow/src/main/java/com/vaadin/flow/component/button/Button.java
+++ b/vaadin-button-flow-parent/vaadin-button-flow/src/main/java/com/vaadin/flow/component/button/Button.java
@@ -30,6 +30,7 @@
import com.vaadin.flow.component.dependency.JsModule;
import com.vaadin.flow.component.dependency.NpmPackage;
import com.vaadin.flow.component.html.Image;
+import com.vaadin.flow.component.page.PendingJavaScriptResult;
import com.vaadin.flow.component.shared.HasPrefix;
import com.vaadin.flow.component.shared.HasSuffix;
import com.vaadin.flow.component.shared.HasThemeVariant;
@@ -57,6 +58,7 @@
@JsModule("@vaadin/polymer-legacy-adapter/style-modules.js")
@NpmPackage(value = "@vaadin/button", version = "24.1.0-alpha5")
@JsModule("@vaadin/button/src/vaadin-button.js")
+@JsModule("./buttonFunctions.js")
public class Button extends Component
implements ClickNotifier