Small fix for header button