Fixed some buttons being disabled

Crude fix for now, will need to be fixed again later if more things are
added. We will need to find a better way to get about this.