diff --git a/booking.import_data.inc b/booking.import_data.inc
index 9d50bc8..a3d8aa6 100644
--- a/booking.import_data.inc
+++ b/booking.import_data.inc
@@ -206,10 +206,10 @@ function booking_import_data_admin_submit($form, &$form_state)
 		}
 		
 		//handle study group processing only if the node id was used as a key
-		if ($key == 'nid') {
+		if (isset($record['nid'])) {
 			$update_text .= _booking_import_studygroup_info($record['nid'], $studygroups_update_array, $studygroup_roles_update_array);
-			drupal_set_message(t("Updating record !nid as follows: !update", array('!nid' => $record['nid'], '!update' => $update_text)));
 		}
+		drupal_set_message(t("Updating record !nid as follows: !update", array('!nid' => $record[$key], '!update' => $update_text)));		
 
 		//$args = $query->getArguments();
 		/*