         ]
     }
 
+    public override func getClassForCollectionKey(_ key: String?) -> AnyClass? {
+        switch key {
+        case String.Key.publishCategories:
+            return PublishOptionItem.self
+        default:
+            return nil
+        }
+    }
+
     public override func didFinishMapping() {
         self.addDate = self.formatDate(from: self.addDateString)
     }


    public override func getClassForObjectKey(_ key: String?) -> AnyClass? {
        switch key {
        case String.Key.publishCategories:
            return PublishOptionItem.self
        default:
            return nil
        }

        static let addDate = "add_date"
        static let dataAccessIsExpired = "data_access_is_expired"
        static let identifier = "id"
        static let neededScopes = "needed_scopes"
        static let publishCategories = "publish_categories"
        static let publishToSocial = "publish_to_social"
        static let thirdPartyUserID = "third_party_user_id"
        static let thirdPartyUserDisplayName = "third_party_user_display_name"
