review comment

Signed-off-by: Rajat Gupta <rgupta@qubole.com>