X-Git-Url: http://git.freeside.biz/gitweb/?a=blobdiff_plain;f=FS%2FFS%2Fpart_referral.pm;h=c94c57e19e7e5fe17424b4c3404f1e133dc9ece8;hb=72deba42ac5847c2a6bdeea20157035b8f9df7ae;hp=87bc87cbac98cd58e4a8eaa1c96fae7c69b5af90;hpb=9509e5bfb7f9331303153cac24d7bfecbe2ea9f1;p=freeside.git

diff --git a/FS/FS/part_referral.pm b/FS/FS/part_referral.pm
index 87bc87cba..c94c57e19 100644
--- a/FS/FS/part_referral.pm
+++ b/FS/FS/part_referral.pm
@@ -1,11 +1,12 @@
 package FS::part_referral;
 
 use strict;
-use vars qw( @ISA );
+use vars qw( @ISA $setup_hack );
 use FS::Record qw( qsearch qsearchs dbh );
 use FS::agent;
 
 @ISA = qw( FS::Record );
+$setup_hack = 0;
 
 =head1 NAME
 
@@ -100,7 +101,10 @@ sub check {
     || $self->ut_text('referral')
     || $self->ut_enum('disabled', [ '', 'Y' ] )
     #|| $self->ut_foreign_keyn('agentnum', 'agent', 'agentnum')
-    || $self->ut_agentnum_acl('agentnum', 'Edit global advertising sources')
+    || ( $setup_hack
+           ? $self->ut_foreign_keyn('agentnum', 'agent', 'agentnum' )
+           : $self->ut_agentnum_acl('agentnum', 'Edit global advertising sources')
+       )
   ;
   return $error if $error;