Fix spec to handle new behavior of identity map