[Fix] potential NPE