moify unit tests