diff --git a/hypopg_table.c b/hypopg_table.c
index 54ff5db..4fb1c9e 100644
--- a/hypopg_table.c
+++ b/hypopg_table.c
@@ -743,11 +743,13 @@ hypo_generate_partkey(CreateStmt *stmt, Oid parentid, hypoTable *entry)
 	 */
 	stmt->partspec = transformPartitionSpec(rel, stmt->partspec,
 											&strategy);
+	key->strategy = strategy;
 
 	ComputePartitionAttrs(rel, stmt->partspec->partParams,
 						  partattrs, &partexprs, partopclass,
 						  partcollation, strategy);
 
+
 	/*--- Adapted from RelationBuildPartitionKey ---*/
 {
 	ListCell   *partexprs_item;