1 files changed, 3 insertions, 3 deletions
diff --git a/classy.py b/classy.py
index 54cb12d..6d23b7b 100644
--- a/classy.py
+++ b/classy.py
@@ -599,10 +599,10 @@ def create_class():
 @menu_item('Delete Class')
 def delete_class():
 	cls = choose_class('Choose the class to delete', False)
-	if base != None:
+	if cls != None:
 		name = cls.name
-		del db.class_by_struct[cls.struct]
-		del db.class_by_name[cls.name]
+		del db.classes_by_struct[cls.struct]
+		del db.classes_by_name[cls.name]
 		print("Deleted class %s" % name)
 	else:
 		Warning("No class chosen")