Fix adding entities to undo lists

Derp.