Fix null attribute error