diff --git a/tests/ext/pymysql/test_pymysql.py b/tests/ext/pymysql/test_pymysql.py
index 9333bef4..0fa3beb1 100644
--- a/tests/ext/pymysql/test_pymysql.py
+++ b/tests/ext/pymysql/test_pymysql.py
@@ -76,4 +76,4 @@ def test_execute_bad_query():
     assert sql['database_version']
 
     exception = subsegment.cause['exceptions'][0]
-    assert exception.type == 'InternalError'
+    assert exception.type is not None