Allow for mode to be set via a proc too