unit test updated