DOC: fix docstring for flag checking

Update docstring for new API.