diff --git a/regal/regal.py b/regal/regal.py
index c12c91d..0e15726 100644
--- a/regal/regal.py
+++ b/regal/regal.py
@@ -129,6 +129,7 @@ def learn_representations(args):
                             normalize = True, 
                             gammastruc = args.gammastruc, 
                             gammaattr = args.gammaattr)
+    rep_method.p = args.dimensions
     if max_layer is None:
         max_layer = 1000
     print("Learning representations with max layer %d and alpha = %f" % (max_layer, alpha))