diff --git a/cromulant/ants.py b/cromulant/ants.py index 081fdac..1ec7fdf 100644 --- a/cromulant/ants.py +++ b/cromulant/ants.py @@ -248,7 +248,8 @@ class Ants: words.extend(Utils.random_word(2 - len(words))) words = [ - word if word.lower() != "of" else Utils.random_word()[0] for word in words + word if word.lower() != "of" else Utils.random_word()[0] + for word in words ] words = [Utils.capitalize(word) for word in words] @@ -266,7 +267,6 @@ class Ants: words_1 = split(ant_1) words_2 = split(ant_2) - words_1 = fill(words_1) words_2 = fill(words_2) diff --git a/cromulant/utils.py b/cromulant/utils.py index 8aa72a8..188ea98 100644 --- a/cromulant/utils.py +++ b/cromulant/utils.py @@ -160,4 +160,4 @@ class Utils: @staticmethod def capitalize(word: str) -> str: - return word[0].upper() + word[1:] \ No newline at end of file + return word[0].upper() + word[1:]