diff --git a/bin/dtram.py b/bin/dtram.py
index b386b61..a4375a1 100644
--- a/bin/dtram.py
+++ b/bin/dtram.py
@@ -192,10 +192,7 @@
     print "#\n#################################### THAT'S IT #####################################\n#"
     print "#\n#                     Thank you for using the pytram package!\n#\n#"
     print "### CITATION\n#"
-    print "#    Statistically optimal analysis of state-discretized trajectory data"
-    print "#    from multiple thermodynamic states;"
-    print "#    Hao Wu, Antonia S.J.S. Mey, Edina Rosta, and Frank Noe"
-    print "#    J. Chem. Phys. 141, 214106 (2014)"
+    dtram_obj.cite( pre="#    " )
     print "#\n####################################################################################\n\n"
 
 
diff --git a/bin/xtram.py b/bin/xtram.py
index 2c9444c..21002f7 100755
--- a/bin/xtram.py
+++ b/bin/xtram.py
@@ -194,9 +194,11 @@
     #   say good bye
     #
     ############################################################################
-    print "#\n###################That's it, now it is time to put the kettle on ##############################\n#"
-    print "#\n#                  Thank you for using xtram in the pytram package!\n#\n#"
-    print "#\n################################################################################################\n\n"
+    print "#\n#################################### THAT'S IT #####################################\n#"
+    print "#\n#                     Thank you for using the pytram package!\n#\n#"
+    print "### CITATION\n#"
+    xtram_obj.cite( pre="#    " )
+    print "#\n####################################################################################\n\n"