X-Git-Url: http://git.freeside.biz/gitweb/?p=freeside.git;a=blobdiff_plain;f=FS%2FFS%2Fcust_bill_pkg_detail.pm;h=4d9ee81919d604ef938165591efbd823bf5c46ac;hp=63d0ac5ba77f39dd02bfb1d39cd1159bec65e3cd;hb=cc9855d8e52fe35a362f7bda01616d53f04a7aab;hpb=62bfeae993beb7f98d50d319360f1fece128982b

diff --git a/FS/FS/cust_bill_pkg_detail.pm b/FS/FS/cust_bill_pkg_detail.pm
index 63d0ac5ba..4d9ee8191 100644
--- a/FS/FS/cust_bill_pkg_detail.pm
+++ b/FS/FS/cust_bill_pkg_detail.pm
@@ -5,6 +5,7 @@ use vars qw( @ISA $me $DEBUG %GetInfoType );
 use HTML::Entities;
 use FS::Record qw( qsearch qsearchs dbdef dbh );
 use FS::cust_bill_pkg;
+use FS::usage_class;
 use FS::Conf;
 
 @ISA = qw(FS::Record);
@@ -240,8 +241,8 @@ sub _upgrade_data { # class method
 
   warn "$me upgrading $class\n" if $DEBUG;
 
-  my $columndef = dbdef->table($class->table)->column('classnum');
-  unless ($columndef->type eq 'int4') {
+  my $type = dbdef->table($class->table)->column('classnum')->type;
+  unless ( $type =~ /^int/i || $type =~ /int$/i ) {
 
     my $dbh = dbh;
     if ( $dbh->{Driver}->{Name} eq 'Pg' ) {