diff --git a/tests/test_cli.py b/tests/test_cli.py
index 89ffb9be2b6c375051298f116b932eb7b8deded6..fe4885eaf3cb8bdb2565ff273f43c89281d156cd 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,21 +1,9 @@
-import sys
-import shlex
-from pathlib import Path
-import subprocess
-
-
-def run_command(cmd: str, args: str):
-    cmd = f"{sys.executable} -m bitia {cmd} {args}"
-    print(f"Executing `{shlex.split(cmd)}`")
-    assert Path(sys.executable).is_file()
-    p = subprocess.run(shlex.split(cmd), capture_output=True)
-    return p.stdout
+from bitia.__main__ import run_pipeline
 
 
 def test_cli_sanity():
-    output = run_command("run", '"ls -ltr /"')
-    print(output.decode().rstrip())
-    assert len(output) > 20, "Too little output"
+    for line in run_pipeline("ls -ltr /"):
+        print(line)
 
 
 if __name__ == "__main__":