diff --git a/frontend/components/flash_messages/FlashMessage/FlashMessage.jsx b/frontend/components/flash_messages/FlashMessage/FlashMessage.jsx
index 5c1f31170b..f95adc9353 100644
--- a/frontend/components/flash_messages/FlashMessage/FlashMessage.jsx
+++ b/frontend/components/flash_messages/FlashMessage/FlashMessage.jsx
@@ -30,9 +30,7 @@ const FlashMessage = ({
 
   // Success alerts will not be visible after 4 seconds
   if (alertType === "success") {
-    setTimeout(function () {
-      document.getElementById(`${klass}`).style.display = "none";
-    }, 4000);
+    setTimeout(onRemoveFlash, 4000);
   }
 
   return (