Fix click listener