Fixed integration tests issue