1 files changed, 1 insertions, 1 deletions
diff --git a/DiskXYZ.py b/DiskXYZ.py
index e2f070f..b593f2d 100644
--- a/DiskXYZ.py
+++ b/DiskXYZ.py
@@ -491,7 +491,7 @@ class DiskReduce(DiskReduceBase):
 
     def compute_Omega_xy(self):
         self.Omega_xy = self.vcyl_xy["vtheta"] / self.rt["r"]
-        self.Omega_xy[self.Omega_xy == 0.0] = np.nan
+        self.Omega_xy[self.Omega_xy <= 1.0e-6] = np.nan
 
     def compute_h_xy(self):
         self.h_xy = np.sqrt(self.cs2_xy) / self.Omega_xy