fix simple button code