Fix small expand/collapse animation bug