diff --git a/tests/test_cli.py b/tests/test_cli.py
index 1c964e7..cffe2ad 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -193,7 +193,7 @@ def _download_book(book_id: int) -> Optional[BytesIO]:
     monkeypatch.setattr(cli, "setup_settings", lambda: None)
     monkeypatch.setattr(cli, "download_book", _download_book)
     monkeypatch.setattr("getpass.getpass", _getpass_mock)
-    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != "5678")
+    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != 5678)
 
     with patch.object(
         sys, "argv", ["gutenberg2kindle", "send", "--book-id", "1234", "5678", "9101"]
@@ -225,7 +225,7 @@ def _download_book(book_id: int) -> Optional[BytesIO]:
     monkeypatch.setattr(cli, "setup_settings", lambda: None)
     monkeypatch.setattr(cli, "download_book", _download_book)
     monkeypatch.setattr("getpass.getpass", _getpass_mock)
-    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != "5678")
+    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != 5678)
 
     with patch.object(
         sys,