diff --git a/generate_tfrecord.py b/generate_tfrecord.py
index 87fa5a98..0ad13386 100644
--- a/generate_tfrecord.py
+++ b/generate_tfrecord.py
@@ -29,7 +29,7 @@
 
 # TO-DO replace this with label map
 def class_text_to_int(row_label):
-    if row_label == 'raccoon':
+    if row_label == 'hand':
         return 1
     else:
         None