update unittest code