upgrade unit tests