diff --git a/ci/check_inference.py b/ci/check_inference.py
index a502c4c..152f463 100644
--- a/ci/check_inference.py
+++ b/ci/check_inference.py
@@ -18,4 +18,4 @@
         print(line.decode(), end="")
     
     print("Asserting that the output contains the expected SQL query")
-    assert b"[SQL] SELECT" in output and b"[/SQL]" in output
+    assert "[SQL] SELECT" in output and "[/SQL]" in output