changed some unit tests