Update unit test and remove log statement