diff --git a/sites/all/themes/unl_wdn/template.php b/sites/all/themes/unl_wdn/template.php
index 15e5d3ae35a2d804e7e163f67020640e97286cf5..8d51c52ad6c29413a6af424c05fed578c75297c6 100644
--- a/sites/all/themes/unl_wdn/template.php
+++ b/sites/all/themes/unl_wdn/template.php
@@ -125,8 +125,10 @@ function unl_wdn_menu_local_task($variables)
     return '<li' . (!empty($variables['element']['#active']) ? ' class="selected"' : '') . '>' . l($link_text, $link['href'], $link['localized_options']) . "</li>\n";
 }
 
-function unl_wdn_status_messages()
+function unl_wdn_status_messages($variables)
 {
+    $display = $variables['display'];
+    
     $output = '';
     foreach (drupal_get_messages($display) as $type => $messages) {
         $type = ucfirst($type);