fix button problem