fix nil class error