diff --git a/tests/test_mega_millions.py b/tests/test_mega_millions.py
index 5c10a44..b1fc4fa 100644
--- a/tests/test_mega_millions.py
+++ b/tests/test_mega_millions.py
@@ -1,7 +1,7 @@
 from datetime import datetime
 import pytest
 
-from sqlite3 import IntegrityError
+from sqlalchemy.exc import IntegrityError
 from httpx import ConnectError
 
 from conftest import MegaMillionsDraw
diff --git a/tests/test_powerball.py b/tests/test_powerball.py
index 3b5c002..af9f252 100644
--- a/tests/test_powerball.py
+++ b/tests/test_powerball.py
@@ -1,7 +1,7 @@
 from datetime import datetime
 import pytest
 
-from sqlite3 import IntegrityError
+from sqlalchemy.exc import IntegrityError
 from httpx import ConnectError
 
 from conftest import PowerballDraw