Add back missing button class