fix for integration test