diff --git a/nnfwtbn/plot.py b/nnfwtbn/plot.py
index 2cd5197b3cfa527cabb09bafdf23eb12893bda5e..6c9fa64d61423120eb9526ebe0364eb8f4df0e59 100644
--- a/nnfwtbn/plot.py
+++ b/nnfwtbn/plot.py
@@ -319,7 +319,7 @@ def hist(dataframe, variable, bins, stacks, selection=None,
     if return_uhepp:
         return uhepp_obj
     else:
-        uhepp_obj.render()
+        return uhepp_obj.render()
 
 def _transpose(array):
     """Return a transposed version of the array"""