diff --git a/tests/test_cli.py b/tests/test_cli.py
index dfe0f09..6a48169 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 "Specify an URL" in reply
+    assert ("waybackpy %s \nSee 'waybackpy --help" % (__version__)) == reply
 
 def test_main():
     # This also tests the parse_args method in cli.py