Removing check on entity type as it can only be a Work now