fix button layout in test blocks