Add check to not dissoc if nil