diff --git a/markov.py b/markov.py index eed84f0..c102691 100755 --- a/markov.py +++ b/markov.py @@ -16,7 +16,7 @@ class Possy(markovify.NewlineText): def sanitizeText(text): split = text.split() try: - if "@" in (text[0], text[1]): + if text[0] == "@" or text[1] == "@": if split[1][0] not in string.ascii_lowercase: return sanitizeText(" ".join(split[1:])) if split[-1][0] == "@":