Fix integration test to handle multiple buttons