diff --git a/umap/parametric_umap.py b/umap/parametric_umap.py
index 95d1bced..51674dcf 100644
--- a/umap/parametric_umap.py
+++ b/umap/parametric_umap.py
@@ -483,7 +483,7 @@ def __getstate__(self):
         return dict(
             (k, v)
             for (k, v) in self.__dict__.items()
-            if should_pickle(k, v) and k not in ("optimizer", "encoder", "decoder", "parametric_model")
+            if k not in ("optimizer", "encoder", "decoder", "parametric_model") and should_pickle(k, v)
         )
 
     def save(self, save_location, verbose=True):