diff --git a/bots/modules/_mtg.py b/bots/modules/_mtg.py index f390c55..2dcb5ec 100644 --- a/bots/modules/_mtg.py +++ b/bots/modules/_mtg.py @@ -49,9 +49,10 @@ class MTG(object): if second_word is not None: return first_word, second_word elif first_word is not None: - if len(self.MC['.' + '+' + first_word]) != 0: + if ('.' + '+' + first_word in self.MC) and len(self.MC['.' + '+' + first_word]) != 0: second_word = random.choice(self.MC['.' + '+' + first_word]) - elif len(self.MC['.' + '+' + first_word.capitalize()]) != 0: + elif ('.' + '+' + first_word.capitalize() in self.MC) and \ + len(self.MC['.' + '+' + first_word.capitalize()]) != 0: second_word = random.choice(self.MC['.' + '+' + first_word.capitalize()]) else: return self.choose_start_words()