diff --git a/amplpy/tests/test_ampl.py b/amplpy/tests/test_ampl.py
index 966f5d3..10856db 100644
--- a/amplpy/tests/test_ampl.py
+++ b/amplpy/tests/test_ampl.py
@@ -195,7 +195,7 @@ def warning(self, exception):
             ampl.eval("X X;")
 
         ampl.set_output_handler(OutputHandlerRaise())
-        with self.assertRaises(RuntimeError):
+        with self.assertRaises(AssertionError):
             ampl.eval("display 1;")
 
     def test_get_output(self):