diff --git a/bin/vhost-add b/bin/vhost-add
index 04cc335..540e2c5 100755
--- a/bin/vhost-add
+++ b/bin/vhost-add
@@ -96,7 +96,7 @@ TMP_FILE="/tmp/vhost-add-$(tr -dc "qwertQWERTasdfgASDFGzxcvbZXCVB" < /dev/urando
 touch "$TMP_FILE"
 chmod 600 "$TMP_FILE"
 template summary > "$TMP_FILE"
-whiptail --textbox "$TMP_FILE" 25 80
+whiptail --textbox "$TMP_FILE" --title "$TITLE - PREVIEW" 25 80
 rm "$TMP_FILE"
 
 CONTINUE="$(form_yes_no -t "$TITLE" -l "Do you confirm?" -n)"
@@ -166,5 +166,5 @@ TMP_FILE="/tmp/vhost-add-$(tr -dc "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS
 touch "$TMP_FILE"
 chmod 600 "$TMP_FILE"
 template result > "$TMP_FILE"
-whiptail --textbox "$TMP_FILE" 25 80
+whiptail --textbox "$TMP_FILE" --title "$TITLE - FINISH" 25 80
 rm "$TMP_FILE"