diff --git a/lib/batch.js b/lib/batch.js
index 68c16ab..43d1a01 100644
--- a/lib/batch.js
+++ b/lib/batch.js
@@ -48,7 +48,7 @@ internals.paginatedRequest = function (request, table, callback) {
         return callback(err);
       }
 
-      request = resp.UnprocessedKeys;
+      request = { RequestItems: resp.UnprocessedKeys };
       responses.push(resp);
 
       return callback();
@@ -56,7 +56,7 @@ internals.paginatedRequest = function (request, table, callback) {
   };
 
   var testFunc = function () {
-    return request !== null && !_.isEmpty(request);
+    return !_.isEmpty(_.get(request, 'RequestItems'));
   };
 
   var resulsFunc = function (err) {