fixed work button issue