update the unit testing