Fix integration test for CLI