diff --git a/config.pbtxt b/config.pbtxt
index 4840f87..1179b70 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -42,12 +42,14 @@ input [
 
   },
   {
-    name: "top_k"
+    name: "top_k-modify"
     data_type: TYPE_INT32
-    dims: [ 1 ]
+    dims: [ 1.123 ]
 
-    optional: true
 
+    reshape {
+      shape: [ 23 ]
+    }
 
   },
   {