fix nil reference error