fix bug after click