Fixed a bug with unhandled null return value