diff --git a/badge.py b/badge.py
index 685dd35..8999e02 100644
--- a/badge.py
+++ b/badge.py
@@ -67,7 +67,7 @@ class BadgePopulation:
 		badge_names = list(badge_names)
 		for badge_name in badge_names:
 			found=False
-			for badge in self.badges[user]:
+			for badge in self.badges.get(user,[]):
 				if badge.name==badge_name: found=True
 			if not found:
 				raise UserDoesntHaveEnoughBadges(f"User {user} does not have a {badge_name}!")
@@ -77,7 +77,7 @@ class BadgePopulation:
 		N = (((1-W)/len([x for x in self.population if x.normal]))+(W/len(self.population)))**-1
 		for badge_name in badge_names:
 			taken = False
-			for badge in self.badges[user]:
+			for badge in self.badges.get(user,[]):
 				if not taken and badge.name==badge_name:
 					taken=True
 					self.badges[user].remove(badge)