Adjusted user assigned code while adding the associations to the UI