diff --git a/tensorflow/core/ops/array_ops.cc b/tensorflow/core/ops/array_ops.cc
index 7b0557c4595..ad673d3235c 100644
--- a/tensorflow/core/ops/array_ops.cc
+++ b/tensorflow/core/ops/array_ops.cc
@@ -1479,12 +1479,9 @@ Status UniqueIdxShapeFn(InferenceContext* c) {
     }
     c->set_output(1, c->Vector(c->Dim(input, axis)));
     return Status::OK();
-  } else {
-    return errors::InvalidArgument(
-        "axis does not support input tensors larger than 1 elements.");
   }
-
-  return Status::OK();
+  return errors::InvalidArgument(
+      "axis does not support input tensors larger than 1 elements.");
 }
 }  // namespace