diff --git a/tests/test_display.py b/tests/test_display.py
index 931d084f..5e33701b 100644
--- a/tests/test_display.py
+++ b/tests/test_display.py
@@ -1,40 +1,43 @@
 import argparse
 import jrnl
-import pytest 
+import pytest
 import mock
 from jrnl.jrnl import _export_journal
+
+
 @pytest.mark.parametrize(
     "export_format",
     [
         "pretty",
         "short",
-    ]
+    ],
 )
-@mock.patch.object(argparse,'Namespace',return_value={"export":None,"filename":None})
-def test_export_format(mock_args,export_format): 
-    
-    test_journal = jrnl.Journal.Journal 
-    mock_args.export = export_format
-    mock_args.filename = "foo.jrnl"
-    with mock.patch('builtins.print') as print_spy, \
-        mock.patch('jrnl.Journal.Journal.pprint') as mock_pprint:
-        _export_journal(mock_args,test_journal)
-    print_spy.call_args_list = mock_pprint
-    
-    
+@mock.patch.object(
+    argparse, "Namespace", return_value={"export": None, "filename": None}
+)
+def test_export_format(mock_args, export_format):
 
-@mock.patch.object(argparse,'Namespace',return_value={"export":None,"filename":None})
-def test_export_plugin(mock_args): 
+    test_journal = jrnl.Journal.Journal
+    mock_args.export = export_format
+    with mock.patch("builtins.print") as print_spy, mock.patch(
+        "jrnl.Journal.Journal.pprint"
+    ) as mock_pprint:
+        _export_journal(mock_args, test_journal)
+    print_spy.call_args_list = mock_pprint
+
+
+@mock.patch.object(
+    argparse, "Namespace", return_value={"export": None, "filename": None}
+)
+def test_export_plugin(mock_args):
     export_format = "markdown"
-    
-    test_journal = jrnl.Journal.Journal 
+
+    test_journal = jrnl.Journal.Journal
     mock_args.export = export_format
     mock_args.filename = "foo.jrnl"
-    with mock.patch('builtins.print') as print_spy, \
-        mock.patch('jrnl.plugins.get_exporter') as mock_get_exporter,\
-        mock.patch('jrnl.Journal.Journal.pprint') as mock_pprint:
-        _export_journal(mock_args,test_journal)
+    with mock.patch("builtins.print") as print_spy, mock.patch(
+        "jrnl.plugins.get_exporter"
+    ) as mock_get_exporter, mock.patch("jrnl.Journal.Journal.pprint") as mock_pprint:
+        _export_journal(mock_args, test_journal)
     print_spy.call_args_list = mock_pprint
     mock_get_exporter.assert_called_once_with(export_format)
-    
-    
\ No newline at end of file