diff --git a/tests/test_data.py b/tests/test_data.py
index 4411299..fa2cdad 100644
--- a/tests/test_data.py
+++ b/tests/test_data.py
@@ -135,5 +135,9 @@ def test_set_data_directory():
         assert _CONFIG["first_run"]
         assert _CONFIG["data_already_downloaded"] is False
 
+        # Now reset it
+        set_data_directory(_DEFAULT_DIRECTORY)
+        assert _CONFIG["data_dir"] == _DEFAULT_DIRECTORY
+
     finally:
         shutil.rmtree(location)