Clean up button module transitions