Add unit reference to assignment