fix integration tests
fix integration tests