diff --git a/src/main/java/cn/octopusyan/alistgui/util/alert/AlertBuilder.java b/src/main/java/cn/octopusyan/alistgui/util/alert/AlertBuilder.java
index 427dab9..47d9ea3 100644
--- a/src/main/java/cn/octopusyan/alistgui/util/alert/AlertBuilder.java
+++ b/src/main/java/cn/octopusyan/alistgui/util/alert/AlertBuilder.java
@@ -9,7 +9,7 @@ import java.util.Optional;
 /**
  * @author octopus_yan
  */
-public class AlertBuilder extends BaseBuilder<Alert, ButtonType> {
+public class AlertBuilder extends BaseBuilder<AlertBuilder, Alert> {
 
     public AlertBuilder(Alert alert, Window owner) {
         super(alert, owner);
@@ -19,7 +19,7 @@ public class AlertBuilder extends BaseBuilder<Alert, ButtonType> {
      * AlertUtil.confirm
      */
     public void show(AlertUtil.OnClickListener listener) {
-        Optional<ButtonType> result = alert.showAndWait();
+        Optional<ButtonType> result = dialog.showAndWait();
         result.ifPresent(r -> listener.onClicked(r.getText()));
     }
 
@@ -27,7 +27,7 @@ public class AlertBuilder extends BaseBuilder<Alert, ButtonType> {
      * AlertUtil.confirm
      */
     public void show(AlertUtil.OnChoseListener listener) {
-        Optional<ButtonType> result = alert.showAndWait();
+        Optional<ButtonType> result = dialog.showAndWait();
         result.ifPresent(r -> {
             if (r == ButtonType.OK) {
                 listener.confirm();