NPE error fix