diff --git a/src/main/kotlin/unibz.cs.semint.kprime/domain/ddl/Schema.kt b/src/main/kotlin/unibz.cs.semint.kprime/domain/ddl/Schema.kt
index a823f93a6a97c6f4587a68e8f15fdefacd707c94..ac679a190f8a59ea1e7c263675f7ce1a2b8dd446 100644
--- a/src/main/kotlin/unibz.cs.semint.kprime/domain/ddl/Schema.kt
+++ b/src/main/kotlin/unibz.cs.semint.kprime/domain/ddl/Schema.kt
@@ -53,7 +53,7 @@ class Schema () {
         return keys[0].source.columns.toSet()
     }
 
-    fun addKey(tableName:String, k:Set<Column>) {
+    fun addKey(tableName:String, k:Set<Column>): Constraint {
         val primaryConstraint = Constraint()
         primaryConstraint.name="pkey_$tableName"
         primaryConstraint.source.table="$tableName"
@@ -61,6 +61,7 @@ class Schema () {
         primaryConstraint.target.columns.addAll(k)
         primaryConstraint.type= Constraint.TYPE.PRIMARY_KEY.name
         constraints().add(primaryConstraint)
+        return primaryConstraint
     }
 
     fun keys(): List<Constraint> {