diff --git a/.gitignore b/.gitignore
index 675b2d8..dba8556 100644
--- a/.gitignore
+++ b/.gitignore
@@ -159,6 +159,7 @@ notebooks/frames
 rubix/**/*.ipynb
 
 rubix/spectra/ssp/templates/fsps.h5
+rubix/spectra/ssp/templates/*
 rubix/spectra/ssp/templates/*.gz
 rubix/spectra/ssp/templates/*fits.gz
 rubix/spectra/cue/cue/*
diff --git a/rubix/telescope/noise/noise.py b/rubix/telescope/noise/noise.py
index 3f7af5b..6559d41 100644
--- a/rubix/telescope/noise/noise.py
+++ b/rubix/telescope/noise/noise.py
@@ -58,8 +58,9 @@ def calculate_S2N(
     nonzero_mask = flux_image > 0
 
     # Calculate the median flux value where the flux is non-zero
-    median_flux = jnp.median(jnp.where(nonzero_mask, flux_image, jnp.nan))
-    median_flux = jnp.nan_to_num(median_flux, nan=0.0)
+    # median_flux = jnp.median(jnp.where(nonzero_mask, flux_image, jnp.nan))
+    # median_flux = jnp.nan_to_num(median_flux, nan=0.0)
+    median_flux = jnp.nanmedian(jnp.where(nonzero_mask, flux_image, jnp.nan))
 
     # Calculate the noise factor
     noise_factor = jnp.sqrt(median_flux) / observation_signal_to_noise