Clean up from button test