test(natural-language-understanding): fix IT

Author: kevinkowa

natural-language-understanding/src/test/java/com/ibm/watson/natural_language_understanding/v1/NaturalLanguageUnderstandingIT.java

@@ -723,21 +723,17 @@ public void testCreateCategoriesModel() throws Exception {
    */
   @Test
   public void testListCategoriesModels() throws Exception {
-    ListCategoriesModelsResponse response = service.listCategoriesModels().execute().getResult();
+    CategoriesModelList response = service.listCategoriesModels().execute().getResult();
     assertNotNull(response.getModels());
 
-    for (CategoriesModelList categoriesModelList : response.getModels()) {
-      if (categoriesModelList.getModels() != null) {
-        for (CategoriesModel categoriesModel : categoriesModelList.getModels()) {
-          if (categoriesModel.getName().startsWith("testString")) {
-            DeleteCategoriesModelOptions deleteCategoriesModelOptions =
-                new DeleteCategoriesModelOptions.Builder()
-                    .modelId(categoriesModel.getModelId())
-                    .build();
-            DeleteModelResults deleteModelResults =
-                service.deleteCategoriesModel(deleteCategoriesModelOptions).execute().getResult();
-          }
-        }
+    for (CategoriesModel categoriesModel : response.getModels()) {
+      if (categoriesModel.getName().startsWith("testString")) {
+        DeleteCategoriesModelOptions deleteCategoriesModelOptions =
+            new DeleteCategoriesModelOptions.Builder()
+                .modelId(categoriesModel.getModelId())
+                .build();
+        DeleteModelResults deleteModelResults =
+            service.deleteCategoriesModel(deleteCategoriesModelOptions).execute().getResult();
       }
     }
   }
@@ -825,6 +821,7 @@ public void testUpdateCategoriesModel() throws Exception {
               .description("newDescription")
               .modelVersion("testString")
               .versionDescription("testString")
+              .modelId(modelId)
               .build();
       CategoriesModel response2 =
           service.updateCategoriesModel(updateCategoriesModelOptionsModel).execute().getResult();