Fixup button state in test.