add null check, fix npe