Minor update unittest