Fix docs.

cc @shengyou @farrrr