fixing little button bug