Fix button linking bug