Fix null reference error