Fix AS integration tests