Bug fix on null message return