Fix the button to handle transformations