Fixed die on click bug