Fix doc block