Button bugfix