Fixed script on button click