diff --git a/bob/bio/face/embeddings/facenet_sanderberg.py b/bob/bio/face/embeddings/facenet_sanderberg.py
index ceaf9a84c31a8a0b81dd16b63fc6b28617eef3eb..ada60509026b3017e4f9d3f17ee375ef08807981 100644
--- a/bob/bio/face/embeddings/facenet_sanderberg.py
+++ b/bob/bio/face/embeddings/facenet_sanderberg.py
@@ -223,7 +223,6 @@ class FaceNetSanderberg(TransformerMixin, BaseEstimator):
     def __setstate__(self, d):
         # Handling unpicklable objects
         self.__dict__ = d
-        self.loaded = False
 
     def __getstate__(self):
         import tensorflow as tf
@@ -236,6 +235,7 @@ class FaceNetSanderberg(TransformerMixin, BaseEstimator):
         d.pop("images_placeholder") if "images_placeholder" in d else None
         d.pop("phase_train_placeholder") if "phase_train_placeholder" in d else None
         tf.compat.v1.reset_default_graph()
+        self.loaded = False
         return d
 
     def _more_tags(self):
diff --git a/bob/bio/face/embeddings/tensorflow_compat_v1.py b/bob/bio/face/embeddings/tensorflow_compat_v1.py
index 0936523f62e6b148afe7d78c21229b1c3c4f4527..5659cb3f1a7e1c1c95923804dde222e7401fa374 100644
--- a/bob/bio/face/embeddings/tensorflow_compat_v1.py
+++ b/bob/bio/face/embeddings/tensorflow_compat_v1.py
@@ -122,8 +122,7 @@ class TensorflowCompatV1(TransformerMixin, BaseEstimator):
 
     def __setstate__(self, d):
         # Handling unpicklable objects
-        self.__dict__ = d
-        self.loaded = False
+        self.__dict__ = d        
 
     def __getstate__(self):
         import tensorflow as tf
@@ -134,6 +133,7 @@ class TensorflowCompatV1(TransformerMixin, BaseEstimator):
         d.pop("input_tensor", None)
         d.pop("embedding", None)
         tf.compat.v1.reset_default_graph()
+        self.loaded = False
         return d
 
     # def __del__(self):