1 files changed, 4 insertions, 1 deletions
diff --git a/jpa_test/timeout.py b/jpa_test/timeout.py
index 52d7d92..d7381c3 100644
--- a/jpa_test/timeout.py
+++ b/jpa_test/timeout.py
@@ -3,10 +3,13 @@ import errno
 import os
 import signal
 
+class TimeoutError(Exception):
+    pass
+
 def timeout(seconds=10, error_message=os.strerror(errno.ETIME)):
     def decorator(func):
         def _handle_timeout(signum, frame):
-            raise Exception(error_message)
+            raise TimeoutError(error_message)
 
         def wrapper(*args, **kwargs):
             signal.signal(signal.SIGALRM, _handle_timeout)