Update Unit Test for mulit unit test