Update to use new flags syntax

/cc @matthew-andrews