Add NPE check