diff --git a/keepassxc-browser/content/pwgen.js b/keepassxc-browser/content/pwgen.js
index a5dc8fe..9bcb263 100644
--- a/keepassxc-browser/content/pwgen.js
+++ b/keepassxc-browser/content/pwgen.js
@@ -224,7 +224,7 @@ kpxcPasswordDialog.trigger = async function() {
 
 kpxcPasswordDialog.showDialog = function(field, icon) {
     if (!kpxcFields.isVisible(field)) {
-        document.body.removeChild(icon);
+        icon.parentNode.removeChild(icon);
         field.removeAttribute('kpxc-password-field');
         return;
     }
diff --git a/keepassxc-browser/content/username-field.js b/keepassxc-browser/content/username-field.js
index c251104..7af53fe 100644
--- a/keepassxc-browser/content/username-field.js
+++ b/keepassxc-browser/content/username-field.js
@@ -112,7 +112,7 @@ UsernameFieldIcon.prototype.createIcon = function(field) {
 
 const iconClicked = async function(field, icon) {
     if (!kpxcFields.isVisible(field)) {
-        document.body.removeChild(icon);
+        icon.parentNode.removeChild(icon);
         field.removeAttribute('kpxc-username-field');
         return;
     }