I believe you'll need to setup container to be able to return an instance of the entity, correct?