fixed reference issue for button