fix button classes