diff --git a/tools/tokamak_grids/gridgen/restrict_psi_range.pro b/tools/tokamak_grids/gridgen/restrict_psi_range.pro
index 7d7097ce..43d53cbd 100644
--- a/tools/tokamak_grids/gridgen/restrict_psi_range.pro
+++ b/tools/tokamak_grids/gridgen/restrict_psi_range.pro
@@ -24,8 +24,8 @@ FUNCTION restrict_psi_range, critical, psi_outer
     inner_sep = 0
   ENDIF ELSE BEGIN
     ; Need to update inner_sep
-    w2 = WHERE(w EQ critical.inner_sep, count)
-    IF count EQ 1 THEN BEGIN
+    w2 = WHERE(w EQ critical.inner_sep, count2)
+    IF count2 EQ 1 THEN BEGIN
       inner_sep = w2[0]
     ENDIF ELSE BEGIN
       w = [critical.inner_sep]