diff --git a/entityqueue.migrate.inc b/entityqueue.migrate.inc
index d6ffe38..ce17230 100644
--- a/entityqueue.migrate.inc
+++ b/entityqueue.migrate.inc
@@ -25,12 +25,12 @@ class MigrateEntityqueueHandler extends MigrateDestinationHandler {
   }
 
   /**
-   * Gets the entityqueues from $row and returns them in an array.
+   * Gets the entityqueues from $entity and returns them in an array.
    */
   protected function getEntityqueue($entity, $row) {
     $migrate_entityqueue = array();
-    if (isset($row->entityqueue) && !empty($row->entityqueue)) {
-      $migrate_entityqueue = $row->entityqueue;
+    if (isset($entity->entityqueue) && !empty($entity->entityqueue)) {
+      $migrate_entityqueue = $entity->entityqueue;
     }
     // If it is not an array already, makes it one now.
     if ($migrate_entityqueue && !is_array($migrate_entityqueue)) {