diff --git a/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td b/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td
index e1a8bf452..90c800ba3 100644
--- a/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td
+++ b/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td
@@ -373,9 +373,8 @@ def TMTensor_TopkOp : TMTensor_Op<"topk",
     std::optional<Value> indices() {
       if (getNumInputs() < 2) {
         return {};
-      } else {
-        return getInputOperand(1)->get();
       }
+      return getInputOperand(1)->get();
     }
     Value outputValues() {
       return getOutputOperand(0)->get();