Remove WithModel::Dsl's dependency on how to setup before/after