From ede251afb34ff0c69c5001ed116a88a6bfd671d9 Mon Sep 17 00:00:00 2001
From: Akash Mahanty <akash3pro@gmail.com>
Date: Fri, 2 Oct 2020 23:10:48 +0530
Subject: [PATCH] update tests

---
 tests/test_cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_cli.py b/tests/test_cli.py
index 6a48169..d2fa4ce 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -90,7 +90,7 @@ def test_args_handler():
 
     args = argparse.Namespace(url=None, version=False)
     reply = cli.args_handler(args)
-    assert ("waybackpy %s \nSee 'waybackpy --help" % (__version__)) == reply
+    assert ("waybackpy %s" % (__version__)) in reply
 
 def test_main():
     # This also tests the parse_args method in cli.py