Change UI button fix