Update referenced_entity to allow strings to be passed through as well as symbols