From 0df4db7bb40d2a414338f74181e3af618ecf6ea8 Mon Sep 17 00:00:00 2001
From: Jakub Roztocil <jakub@roztocil.co>
Date: Mon, 16 Sep 2019 13:28:01 +0200
Subject: [PATCH] Cleanup

---
 tests/test_httpie.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_httpie.py b/tests/test_httpie.py
index 430459d5..cd634c0c 100644
--- a/tests/test_httpie.py
+++ b/tests/test_httpie.py
@@ -1,7 +1,7 @@
 """High-level tests."""
 import pytest
 
-import httpie.status
+from httpie.status import ExitStatus
 from httpie.cli.exceptions import ParseError
 from utils import MockEnvironment, http, HTTP_OK
 from fixtures import FILE_PATH, FILE_CONTENT
@@ -11,19 +11,19 @@ import httpie
 
 def test_debug():
     r = http('--debug')
-    assert r.exit_status == httpie.status.ExitStatus.SUCCESS
+    assert r.exit_status == ExitStatus.SUCCESS
     assert 'HTTPie %s' % httpie.__version__ in r.stderr
 
 
 def test_help():
     r = http('--help', tolerate_error_exit_status=True)
-    assert r.exit_status == httpie.status.ExitStatus.SUCCESS
+    assert r.exit_status == ExitStatus.SUCCESS
     assert 'https://github.com/jakubroztocil/httpie/issues' in r
 
 
 def test_version():
     r = http('--version', tolerate_error_exit_status=True)
-    assert r.exit_status == httpie.status.ExitStatus.SUCCESS
+    assert r.exit_status == ExitStatus.SUCCESS
     # FIXME: py3 has version in stdout, py2 in stderr
     assert httpie.__version__ == r.strip()