diff --git a/example.yaml b/example.yaml
index c3358c40ce07231ef25009c3992f558dd09c31a2..69e5cb78f6e5a958b8bd2500c28a736f91d2c01c 100644
--- a/example.yaml
+++ b/example.yaml
@@ -6,8 +6,8 @@ embed_model_data: 1
 models:
   preview_net:
     model_file_path: path/to/model64.pb
-    input_node: input
-    output_node: softmax/Reshape_1
+    input_node: input_node
+    output_node: output_node
     input_shape: 1,64,64,3
     output_shape: 1,64,64,2
     runtime: gpu
@@ -16,7 +16,7 @@ models:
   capture_net:
     model_file_path: path/to/model256.pb
     input_node: input_node
-    output_node: softmax/Reshape_1
+    output_node: output_node
     input_shape: 1,256,256,3
     output_shape: 1,256,256,2
     runtime: gpu