fixed Unit Testing(cont)