diff --git a/tests/test_display.py b/tests/test_display.py
index 5e33701b..9cce277a 100644
--- a/tests/test_display.py
+++ b/tests/test_display.py
@@ -1,7 +1,7 @@
 import argparse
 import jrnl
 import pytest
-import mock
+from unittest import mock
 from jrnl.jrnl import _export_journal