diff --git a/test01/pipeline.yaml b/test01/pipeline.yaml
index b635dae..fa0a6c1 100644
--- a/test01/pipeline.yaml
+++ b/test01/pipeline.yaml
@@ -25,14 +25,11 @@ spec:
                     name: http
                     protocol: TCP
       graph:
-        endpoint:
-          type: REST
         name: model-b
-        type: TRANSFORMER
+        type: MODEL
         children:
-          - endpoint:
-              type: REST
+          - name: model-a
             type: MODEL
-            name: model-a
+            children: []
       name: pipeline-predictor
       replicas: 1