Fix null object referece