Fixed missing button when statement change
Merge pull request #2436 from gittip/statement-change-bug

Fixed missing button when statement change