diff --git a/src/solutions/nonlinear_solutions.jl b/src/solutions/nonlinear_solutions.jl
index 32aabde27..88378cc3f 100644
--- a/src/solutions/nonlinear_solutions.jl
+++ b/src/solutions/nonlinear_solutions.jl
@@ -23,7 +23,7 @@ function build_solution(prob::AbstractNonlinearProblem,
                         kwargs...)
 
   T = eltype(eltype(u))
-  N = length((size(prob.u0)...,))
+  N = ndims(u)
 
   NonlinearSolution{T,N,typeof(u),typeof(resid),
                     typeof(prob),typeof(alg),typeof(original),typeof(left)}(