diff --git a/assistant.py b/assistant.py
index ec41920..94bf203 100755
--- a/assistant.py
+++ b/assistant.py
@@ -97,8 +97,7 @@ def highlight_code(language_name, code):
                 lexer = get_lexer_by_name('bash')
     else:
         # If no language is specified, guess the lexer
-        print("LEXER NAME " + lexer_name)
-        lexer = guess_lexer(code.split('\n')[1:-1])
+        lexer = guess_lexer('\n'.join(code.split('\n')[1:-1]))
         if not lexer:
             # If no lexer is guessed, default to bash
             lexer = get_lexer_by_name('bash')