fix nil bugs