X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?p=dbsrgits%2FDBIx-Class-DeploymentHandler.git;a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FDeploymentHandler%2FCookbook%2FCustomResultSource.pod;fp=lib%2FDBIx%2FClass%2FDeploymentHandler%2FCookbook%2FCustomResultSource.pod;h=42cf2d0c64400c4d42890b05ccba182fd0102711;hp=2f1b79d66699c349b413c23e084c3ace96bbd82d;hb=8bb2f1ac7ad65531f27e24967076fa133f08da93;hpb=fa34f4a874631fa68216dca5332c264de9b4753a

diff --git a/lib/DBIx/Class/DeploymentHandler/Cookbook/CustomResultSource.pod b/lib/DBIx/Class/DeploymentHandler/Cookbook/CustomResultSource.pod
index 2f1b79d..42cf2d0 100644
--- a/lib/DBIx/Class/DeploymentHandler/Cookbook/CustomResultSource.pod
+++ b/lib/DBIx/Class/DeploymentHandler/Cookbook/CustomResultSource.pod
@@ -72,17 +72,19 @@ table being set on the original result.
  sub prepare_version_storage_install {
    my $self = shift;
 
-   $self->prepare_resultsource_install(
-     $self->version_storage->version_rs->result_source
-   );
+   $self->prepare_resultsource_install({
+       result_source => $self->version_storage->version_rs->result_source
+   });
  }
 
  sub install_version_storage {
    my $self = shift;
 
-   $self->install_resultsource(
-     $self->version_storage->version_rs->result_source
-   );
+   my $version = (shift || {})->{version} || $self->schema_version;
+   $self->install_resultsource({
+       result_source => $self->version_storage->version_rs->result_source,
+       version       => $version,
+   });
  }
 
  sub prepare_install {