Update tests to correctly load the container with an entity