Fixed up PK::Auto::* to use result_source, added connection, connect and clone method...

[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / PK / Auto / DB2.pm

diff --git a/lib/DBIx/Class/PK/Auto/DB2.pm b/lib/DBIx/Class/PK/Auto/DB2.pm
index 29973d2..a9732d5 100644 (file)
--- a/lib/DBIx/Class/PK/Auto/DB2.pm
+++ b/lib/DBIx/Class/PK/Auto/DB2.pm
@@ -11,7 +11,7 @@ sub last_insert_id
 {
     my ($self) = @_;
 
-    my $dbh = $self->storage->dbh;
+    my $dbh = $self->result_source->storage->dbh;
     my $sth = $dbh->prepare_cached("VALUES(IDENTITY_VAL_LOCAL())");
     $sth->execute();