diff --git a/mindspore/lite/src/lite_session.cc b/mindspore/lite/src/lite_session.cc
index 7f5e43d2c6..d3d4b1f110 100644
--- a/mindspore/lite/src/lite_session.cc
+++ b/mindspore/lite/src/lite_session.cc
@@ -199,7 +199,7 @@ void LiteSession::InitGraphInputMap(const lite::Model *model) {
     MS_ASSERT(in_node != nullptr);
     auto in_size = in_node->input_indices_.size();
     for (size_t i = 0; i < in_size; ++i) {
-      MS_ASSERT(this->input_map_.find(in_node->name_) + std::to_string(i) == this->input_map_.end());
+      MS_ASSERT(this->input_map_.find(in_node->name_ + std::to_string(i)) == this->input_map_.end());
       auto in_tensor_index = size_t(in_node->input_indices_[i]);
       bool is_graph_input = false;
       for (size_t j = 0; j < graph_in_size; ++j) {
diff --git a/mindspore/lite/test/models_onnx.cfg b/mindspore/lite/test/models_onnx.cfg
index 2ce2c6d825..ce7db58afa 100644
--- a/mindspore/lite/test/models_onnx.cfg
+++ b/mindspore/lite/test/models_onnx.cfg
@@ -32,3 +32,5 @@ pointilism-9.onnx
 rain-princess-9.onnx
 udnie-9.onnx
 adversarial_pruning.onnx
+residual_distill_res34_cifar10_bs_1_update.onnx
+residual_distill_res50_cifar10_bs_1_update.onnx