Fixing missing button