diff --git a/uc_restrict_qty.module b/uc_restrict_qty.module
index 1591c61..49dac69 100644
--- a/uc_restrict_qty.module
+++ b/uc_restrict_qty.module
@@ -315,10 +315,10 @@ function uc_restrict_qty_count($form_values) {
 /**
  * Delete all data associated with a given node.
  *
- * @param $nid
- *   A Drupal node ID.
+ * @param obejct $node
+ *   A Drupal node object.
  */
-function uc_restrict_qty_node_delete($nid) {
+function uc_restrict_qty_node_delete($node) {
   db_query("DELETE FROM {uc_restrict_qty_products} WHERE nid = :nid", array(':nid' => $node->nid));
 }