diff --git a/mysql_batch/insert.py b/mysql_batch/insert.py
index 5ca8392..77c72d8 100644
--- a/mysql_batch/insert.py
+++ b/mysql_batch/insert.py
@@ -29,7 +29,7 @@ def insert_into_restaurants(cursor, restaurant):
     cursor.execute(insert_query, (
       restaurant['id'],
       restaurant['name'], 
-      restaurant['custom_category'], 
+      restaurant['category'],
       0, 
       0, 
       restaurant['address'],