Fix integration spec
Fix integrationt test