diff --git a/onnx_mxnet/import_onnx.py b/onnx_mxnet/import_onnx.py
index <HASH>..<HASH> 100644
--- a/onnx_mxnet/import_onnx.py
+++ b/onnx_mxnet/import_onnx.py
@@ -215,13 +215,13 @@ class GraphProto(object):
         MXNet doesnt have a squeeze operator.
         Using "split" to perform similar operation.
         "split" can be slower compared to "reshape".
-        Remove this implementation once mxnet adds the support.
+         This can have performance impact.
+         TODO: Remove this implementation once mxnet adds the support.
         """
         axes = new_attr.get('axis')
         op = mx.sym.split(inputs[0], axis=axes[0], num_outputs=1, squeeze_axis=1)
-        if len(axes) > 1:
-            for i in axes[1:]:
-                op = mx.sym.split(op, axis=i-1, num_outputs=1, squeeze_axis=1)
+        for i in axes[1:]:
+            op = mx.sym.split(op, axis=i-1, num_outputs=1, squeeze_axis=1)
         return op
 
     def _fix_gemm(self, op_name, inputs, old_attr):
Nits: Removed redundant code
Added few comments