Fixed broken reference to Button