Index: flag_content.module
===================================================================
RCS file: /cvs/drupal-contrib/contributions/modules/flag_content/flag_content.module,v
retrieving revision 1.6
diff -u -p -r1.6 flag_content.module
--- flag_content.module	18 Feb 2007 04:10:17 -0000	1.6
+++ flag_content.module	18 Feb 2007 23:54:48 -0000
@@ -136,8 +136,7 @@ function flag_content_link($type, $node 
   return $links;
 }
 
-function flag_content_add() {
-  $nid = (int)arg(2);
+function flag_content_add($nid = 0) {
   if ($nid) {
     // Insert the data into the table
     db_query("INSERT INTO {flag_content} (nid, timestamp) VALUES (%d, %d)", $nid, time());
@@ -151,8 +150,7 @@ function flag_content_add() {
   drupal_goto("node/$nid");
 }  
 
-function flag_content_unflag() {
-  $nid = (int)arg(2);
+function flag_content_unflag($nid = 0) {
   if ($nid) {
     _flag_content_unflag($nid);
     drupal_set_message(t('The item was unflagged.'));