diff --git a/node_revision_restrict.module b/node_revision_restrict.module
index dc1327a..af2465e 100755
--- a/node_revision_restrict.module
+++ b/node_revision_restrict.module
@@ -128,19 +128,20 @@ function node_revision_restrict_form_submit($form, &$form_state) {
  */
 function node_revision_restrict_node_presave($node) {
   $node_type = $node->type;
-  $nid = $node->nid;
-  $node_data = db_query('SELECT * FROM {node} WHERE nid = :nid', array(':nid' => $nid));
-  foreach ($node_data as $node) {
-    $revision_data = node_revision_list($node);
-    foreach ($revision_data as $revision) {
-      $revision_ids[] = $revision->vid;
-    }
-    $count_to_restrict_set_value = variable_get('restrict_node_revision_number_for_' . $node_type);
-    if (isset($count_to_restrict_set_value)) {
+  $count_to_restrict_set_value = variable_get('restrict_node_revision_number_for_' . $node_type);
+  if (isset($count_to_restrict_set_value)) {
+    $nid = $node->nid;
+    $node_data = db_query('SELECT * FROM {node} WHERE nid = :nid', array(':nid' => $nid));
+    foreach ($node_data as $node) {
+      $revision_data = node_revision_list($node);
+      foreach ($revision_data as $revision) {
+        $revision_ids[] = $revision->vid;
+      }
+      
       $node_to_restrict = array_slice($revision_ids, $count_to_restrict_set_value, count($revision_ids));
       foreach ($node_to_restrict as $revision_id) {
         node_revision_delete($revision_id);
-      }
+      }  
     }
   }
 }