fix integration tests issue