diff --git a/hpvm/projects/keras/frontend/promise_translator.py b/hpvm/projects/keras/frontend/promise_translator.py
index f655174c746fbfdac01f10ca344317e5cd27f7d5..ee33b1c18ad66095aaba463952e8de8d84e0e647 100644
--- a/hpvm/projects/keras/frontend/promise_translator.py
+++ b/hpvm/projects/keras/frontend/promise_translator.py
@@ -306,7 +306,7 @@ class PromiseRtTranslator:
           layer_str += op.activation_type + " fp32 1 "
 
       if op_type == "DepthwiseConv2D":
-        layer_str += "depthwise_conv fp32 1"
+        layer_str += "group_conv fp32 1"
         if op.use_bias:
           layer_str += "add "
         if op.activation_type != "linear":
@@ -326,7 +326,7 @@ class PromiseRtTranslator:
       if op_type == "MaxPooling2D":
         layer_str += "pool_max fp32 1 "
 
-      if op_type == "AvgPooling2D":
+      if op_type == "AveragePooling2D":
         layer_str += "pool_mean fp32 1 "
       
       if op_type == "Add":    
@@ -570,6 +570,9 @@ class PromiseRtTranslator:
     first_op = state.getFirstOp()
     last_op = state.getFirstOp()
 
+    input_var = self.getSingleInputName(first_op)
+    output_var = self.getVariableName(last_op)
+
     self.appendLayerString("BatchNorm", state)
 
     state.clear()
@@ -637,8 +640,7 @@ class PromiseRtTranslator:
     
   # FIXME: Only supporting single AveragePooling layers
   def genPoolLayer(self, state):
-    print ("\n\n Layer = ", state.op_string, "\n\n")
-
+  
     # For single pool layer should be all same
     pool_op = state.getPoolOp()