Good to see you guys already nabbed this one, this issue just tripped one of my tests during my upgrade validation. It looks like the behaviour where if you look up an entity by either its root class or explicit type that you should get a hit like it was prior to