Relations removed in AfterSaveEntities instead of SaveChanges

I don't know if we're doing things  the way it should be but here's what we do:

1. In AfterSaveEntities, we need to work with entities and their associations. 
2. In AfterSaveEntities, we re-use the nhibernate session to update some entities.

We can't do those things because associations are removed in the SaveChangesCore, before calling our delegate.

I've done a workaroud which works well for us. Do you think it's a viable solution?