Update missed integration test