unit test update after fixed integration