Fix stochastic singularity check with `heteroskedastic_shocks`