I think I actually had a similar problem, but using `filter`. Don't really remember though.