Fixing allclose bug w/ infinite values