change welcome message also changes the logout text

Hi,

I am using TML and I think it great! but I am having a issue with the “welcome, user” message on the widget. It’s been posted by the plugin’s author that if you use the code below, you can change the welcome message. It really changes the greeting message, but it also changes the “logout” text (just the text, the link still works) with the greeting message.

I don’t know how to fix it. Any ideas?

<?php

function tml_title( $title, $action ) {
	if ( is_user_logged_in() )
		$title = 'YOUR NEW MESSAGE';
	return $title;
}
add_filter( 'tml_title', 'tml_title', 10, 2 );

https://wordpress.org/plugins/theme-my-login/