Fixes doc issue, thanks @sandeepankundu