fix little buge