diff --git a/tests/test_requests_html.py b/tests/test_requests_html.py
index 12c7ef6..540a6ee 100644
--- a/tests/test_requests_html.py
+++ b/tests/test_requests_html.py
@@ -1,9 +1,9 @@
 import os
 
-from requests_html import Session
+from requests_html import HTMLSession
 from requests_file import FileAdapter
 
-session = Session()
+session = HTMLSession()
 session.mount('file://', FileAdapter())