diff --git a/packages/syft/src/syft/service/dataset/dataset.py b/packages/syft/src/syft/service/dataset/dataset.py
index 185774fa86c..c9b8137aadf 100644
--- a/packages/syft/src/syft/service/dataset/dataset.py
+++ b/packages/syft/src/syft/service/dataset/dataset.py
@@ -306,7 +306,7 @@ def _is_action_data_empty(obj: Any) -> bool:
 
 
 def check_mock(data: Any, mock: Any) -> bool:
-    if type(data) == type(mock):
+    if type(data) is type(mock):
         return True
 
     return _is_action_data_empty(mock) or _is_action_data_empty(data)