Fixed button js bug