fixng button issue