diff --git a/src/vendor-no-composer/robinthehood/ModifiedStdModule/Classes/StdModule.php b/src/vendor-no-composer/robinthehood/ModifiedStdModule/Classes/StdModule.php
index e4bec1f..4228235 100644
--- a/src/vendor-no-composer/robinthehood/ModifiedStdModule/Classes/StdModule.php
+++ b/src/vendor-no-composer/robinthehood/ModifiedStdModule/Classes/StdModule.php
@@ -329,13 +329,9 @@ protected function addMessage(string $message, int $messageType = self::MESSAGE_
             echo '<br>';
         }
 
-        if ($messageType == self::MESSAGE_ERROR) {
-            $class = 'error_message';
-        } elseif ($messageType == self::MESSAGE_SUCCESS) {
-            $class = 'success_message';
-        } else {
-            $class = 'error_message';
-        }
+        $class = self::MESSAGE_SUCCESS === $messageType
+               ? 'success_message'
+               : 'error_message';
 
         if (!self::$messages[$hash]) {
             echo '<div class="' . $class . '">' . $message . '</div>';