fix nil bug