fix little expand/collapse bug