diff --git a/tfjs-core/src/ops/step.ts b/tfjs-core/src/ops/step.ts
index b279a2b899..26c9100262 100644
--- a/tfjs-core/src/ops/step.ts
+++ b/tfjs-core/src/ops/step.ts
@@ -26,7 +26,7 @@ import {TensorLike} from '../types';
 import {op} from './operation';
 
 /**
- * Computes step of the input `tf.Tensor` element-wise: `x > 0 ? 1 : alpha * x`
+ * Computes step of the input `tf.Tensor` element-wise: `x > 0 ? 1 : alpha`
  *
  * ```js
  * const x = tf.tensor1d([0, 2, -1, -3]);
@@ -34,7 +34,7 @@ import {op} from './operation';
  * x.step(.5).print();  // or tf.step(x, .5)
  * ```
  * @param x The input tensor.
- * @param alpha The gradient when input is negative.
+ * @param alpha The gradient when input is negative. Defaults to 0.
  *
  * @doc {heading: 'Operations', subheading: 'Basic math'}
  */