Fix IntegrationTest