diff --git a/docs/callback.rst b/docs/callback.rst
index df6abd3..799a7c9 100644
--- a/docs/callback.rst
+++ b/docs/callback.rst
@@ -29,7 +29,7 @@ initialization.
 .. testcode::
 
     from kafka import KafkaConsumer
-    from kq import Worker
+    from kq import Worker, Job
 
 
     def callback(status, message, job, result, exception, stacktrace):