Fixed problem with the adding of the new buttons