fix wrong message at the end.