[BUGFIX] Fix buttons