Fixing click listener reference