hacky fix for integration test