update some unit tests