diff --git a/src/kbmod/filters/sigma_g_filter.py b/src/kbmod/filters/sigma_g_filter.py
index 6ab42d48..87cdd67e 100644
--- a/src/kbmod/filters/sigma_g_filter.py
+++ b/src/kbmod/filters/sigma_g_filter.py
@@ -182,7 +182,7 @@ def compute_clipped_sigma_g_matrix(self, lh):
             A N x T matrix of Booleans indicating if each point is valid (True)
             or has been filtered (False).
         """
-        inds_valid = self.sigma_g_jax_fn(jnp.asarray(lh))
+        inds_valid = self.sigma_g_jax_fn(jnp.asarray(lh)).block_until_ready()
         return inds_valid