diff --git a/src/filtrations/custom.jl b/src/filtrations/custom.jl
index 680d033..56d5331 100644
--- a/src/filtrations/custom.jl
+++ b/src/filtrations/custom.jl
@@ -122,7 +122,7 @@ end
         idx = index(vs)
         d_dict[idx] = min(birth, get(d_dict, idx, typemax(T)))
     end
-    _insert_simplex_facets!(dicts, vertices, birth, Val(N - 1))
+    return _insert_simplex_facets!(dicts, vertices, birth, Val(N - 1))
 end
 
 function _adjacency_matrix(dicts)
diff --git a/test/aqua.jl b/test/aqua.jl
index eb78cf9..d2bbbd8 100644
--- a/test/aqua.jl
+++ b/test/aqua.jl
@@ -1,5 +1,6 @@
 using Aqua
 using Ripserer
+using PersistenceDiagrams
 
 Aqua.test_all(Ripserer; ambiguities=false, piracies=false)
-Aqua.test_piracies(Ripserer; treat_as_own=PersistenceDiagram)
+Aqua.test_piracies(Ripserer; treat_as_own=[PersistenceDiagram,PersistenceInterval])