Fixed issue when passing in entityName

Needed to add if statement to allow for passing in a entity name that wasn't the virtual entity service's entity name.