diff --git a/tests/test_cmds.py b/tests/test_cmds.py
index 7c155c2..0751636 100644
--- a/tests/test_cmds.py
+++ b/tests/test_cmds.py
@@ -1,7 +1,7 @@
 import pytest
 from click.testing import CliRunner
 
-from imaginairy.cmds import imagine_cmd
+from imaginairy.cmds import edit_image, imagine_cmd
 from imaginairy.utils import get_device
 from tests import TESTS_FOLDER
 
@@ -22,3 +22,19 @@ def test_imagine_cmd():
         ],
     )
     assert result.exit_code == 0
+
+
+@pytest.mark.skipif(get_device() == "cpu", reason="Too slow to run on CPU")
+def test_edit_cmd():
+    runner = CliRunner()
+    result = runner.invoke(
+        edit_image,
+        [
+            f"{TESTS_FOLDER}/data/dog.jpg",
+            "--steps",
+            "1",
+            "-p",
+            "turn the dog into a cat",
+        ],
+    )
+    assert result.exit_code == 0