diff --git a/domain_source/domain_source.module b/domain_source/domain_source.module
index 4ec6298..baaf457 100644
--- a/domain_source/domain_source.module
+++ b/domain_source/domain_source.module
@@ -312,10 +312,6 @@ function domain_source_lookup($nodes, $domains = array(), $domain_site = FALSE,
       $domain_site = $lookup['domain_site'];
       $domains = $lookup['domain_id'];
     }
-    // If no valid source is found, take the first match from the $domains array.
-    if (empty($domain_site) && !empty($domains) && !in_array($source, $domains)) {
-      $source = current($domains);
-    }
     // If set to 'Use active domain', return that status flag.
     if ($source == DOMAIN_SOURCE_USE_ACTIVE) {
       $lookups[$data->nid] = $source;
@@ -327,6 +323,10 @@ function domain_source_lookup($nodes, $domains = array(), $domain_site = FALSE,
         $lookups[$data->nid] = $domain['domain_id'];
       }
     }
+    // If no valid source is found, take the first match from the $domains array.
+    if (empty($domain_site) && !empty($domains) && !in_array($source, $domains)) {
+      $source = current($domains);
+    }
   }
   // Make sure we found something for every requested node.
   $no_records = $node_ids;