Fix work buttons