--- a/weight/probability/ad_weight_probability.module	Mon Apr 20 09:49:37 2009 +0000
+++ b/weight/probability/ad_weight_probability.module	Mon Apr 27 10:27:29 2009 +0000
@@ -1,5 +1,5 @@
 <?php
-// $Id: ad_weight_probability.module,v 1.1.4.5 2009/03/27 19:11:50 jeremy Exp $
+// $Id: ad_weight_probability.module,v 1.1.4.6 2009/04/22 15:14:46 jeremy Exp $
 
 /**
  * @file
@@ -48,13 +48,12 @@
     case 'insert':
     case 'update':
       if (user_access('configure ad probability')) {
-        // Fully load the node object to confirm that we are working with an
-        // advertisement.
-        $ad = node_load($node->nid);
-        if (isset($ad->adtype)) {
+        if (is_object($node) && isset($node->adtype) &&
+            isset($node->probability) && isset($node->nid)) {
           return _ad_weight_probability_node_save($node, $op);
         }
       }
+      break;
     case 'delete':
       return _ad_weight_probability_node_delete($node);
   }
@@ -99,7 +98,7 @@
  * Helper function, save the probability to the database.
  */
 function _ad_weight_probability_node_save($node) {
-  if (is_object($node) && $node->nid) {
+  if (isset($node->nid) && $node->nid) {
     db_query('UPDATE {ad_weight_probability} SET probability = %d WHERE aid = %d', $node->probability, $node->nid);
     if (!db_affected_rows()) {
       db_query('INSERT INTO {ad_weight_probability} (aid, probability) VALUES(%d, %d)', $node->nid, $node->probability);