diff --git a/markov.py b/markov.py index 9788f54..eed84f0 100755 --- a/markov.py +++ b/markov.py @@ -18,7 +18,7 @@ def sanitizeText(text): try: if "@" in (text[0], text[1]): if split[1][0] not in string.ascii_lowercase: - return sanitizeText(text.partition(" ")[2]) + return sanitizeText(" ".join(split[1:])) if split[-1][0] == "@": return sanitizeText(" ".join(split[:-1])) if text[:4] == "RT @":