diff --git a/src/assign_images.py b/src/assign_images.py
index 139f784..09045e6 100644
--- a/src/assign_images.py
+++ b/src/assign_images.py
@@ -97,7 +97,7 @@ def main(
     log.info(gdf.head())
     log.info(
         "Are There Duplicates? {}",
-        gdf[gdf["image_id"] is not None and gdf["image_id"] != np.nan]["image_id"]
+        gdf[gdf["image_id"] is not None and gdf["image_id"].notna()]["image_id"]
         .duplicated()
         .any(),
     )