Fixed missed button after previous commit