Also, should probably just be setting `$context['message']`, instead of `drupal_set_message()`.