fix null reference error