Add support for using controllers in dialog fragments