From 2b408a26fc10eca3e45f30e8bc2d1a11594ee140 Mon Sep 17 00:00:00 2001
From: Lucas Pereira <lucas.lmspereira@gmail.com>
Date: Fri, 8 Dec 2023 12:44:28 -0300
Subject: [PATCH] fix (#168)

---
 src/CheqFilters/density_filter.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/CheqFilters/density_filter.jl b/src/CheqFilters/density_filter.jl
index 4df569da..3276dbda 100644
--- a/src/CheqFilters/density_filter.jl
+++ b/src/CheqFilters/density_filter.jl
@@ -43,10 +43,10 @@ function DensityFilter(
 end
 
 function (cf::DensityFilter{true,T})(x::PseudoDensities{I,P}) where {I,P,T}
-    cf.rmin <= 0 && return x
+    cf.rmin <= 0 && return PseudoDensities{I,P,true}(x.x)
     @unpack jacobian = cf
     out = similar(x.x)
-    mul!(out, jacobian, x)
+    mul!(out, jacobian, x.x)
     return PseudoDensities{I,P,true}(out)
 end
 function ChainRulesCore.rrule(f::DensityFilter{true}, x::PseudoDensities)