diff --git a/config.pbtxt b/config.pbtxt
index 1179b70..23af029 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -44,7 +44,7 @@ input [
   {
     name: "top_k-modify"
     data_type: TYPE_INT32
-    dims: [ 1.123 ]
+    dims: [ 1 ]
 
 
     reshape {