diff --git a/src/expectation.jl b/src/expectation.jl
index 65929e8..6c57a8b 100644
--- a/src/expectation.jl
+++ b/src/expectation.jl
@@ -226,8 +226,7 @@ function integrate(quadalg, adalg::AbstractExpectationADAlgorithm, f, lb::TB, ub
                    nout = nothing, batch = nothing,
                    kwargs...) where {TB}
     #TODO check batch iip type stability w/ IntegralProblem{XXXX}
-    batch = batch==0 ? nothing : batch
-    prob = IntegralProblem(f, lb, ub, p; nout = nout, batch = batch)
+    prob = IntegralProblem(f, (lb, ub), p; nout = nout, batch = batch)
     solve(prob, quadalg; kwargs...)
 end