Fix null coal assign (with less code duplication)

Fix null coal assign (with less code duplication) #7331