X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FProducer%2FDiaUml.pm;fp=lib%2FSQL%2FTranslator%2FProducer%2FDiaUml.pm;h=7c02d33a20f994fb0920c5631e7908a80686e072;hb=6e64adbe068b34d9fdad49f6bcd02fb265ca0bce;hp=ea985f12e5b263163a170a6f57d7dec922107e07;hpb=c734e9e6177653db3e0248f21c5fae636c5b03ed;p=dbsrgits%2FSQL-Translator.git

diff --git a/lib/SQL/Translator/Producer/DiaUml.pm b/lib/SQL/Translator/Producer/DiaUml.pm
index ea985f1..7c02d33 100644
--- a/lib/SQL/Translator/Producer/DiaUml.pm
+++ b/lib/SQL/Translator/Producer/DiaUml.pm
@@ -60,17 +60,15 @@ use vars qw[ $DEBUG $VERSION @EXPORT_OK ];
 $VERSION = '1.59';
 $DEBUG   = 0 unless defined $DEBUG;
 
+use File::ShareDir qw/dist_dir/;
+
 use SQL::Translator::Utils 'debug';
 use base qw/SQL::Translator::Producer::TT::Base/;
 # Convert produce call into a method call on our class
 sub produce { return __PACKAGE__->new( translator => shift )->run; };
 
-# Uses dir in lib with this mods name as the template dir
-my $_TEMPLATE_DIR = __FILE__;
-$_TEMPLATE_DIR =~ s/\.pm$//;
-
 sub tt_config {
-    ( INCLUDE_PATH => $_TEMPLATE_DIR );
+    ( INCLUDE_PATH => File::Spec->catdir (dist_dir('SQL-Translator'), 'DiaUml') );
 }
 
 sub tt_schema { 'schema.tt2' }