Bugfix in non-linear initial conditions