return values when using native traits

[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Object / Index.pm

diff --git a/lib/SQL/Translator/Object/Index.pm b/lib/SQL/Translator/Object/Index.pm
index f5729a0..dbca752 100644 (file)
--- a/lib/SQL/Translator/Object/Index.pm
+++ b/lib/SQL/Translator/Object/Index.pm
@@ -40,7 +40,10 @@ class SQL::Translator::Object::Index extends SQL::Translator::Object {
         default => 'NORMAL',
     );
 
-    around add_column(Column $column) { $self->$orig($column->name, $column) }
+    around add_column(Column $column) {
+        $self->$orig($column->name, $column);
+        return $self->get_column($column->name);
+    }
 
     method is_valid { $self->table && scalar $self->get_columns ? 1 : undef }
 }