fixed button reference and code to re-enable it