I just checked that using the clip kw argument (with (0, 1)) improves the result. It could be interesting to display the result for the clipped solution as well (no need for an additional figure, just print the reconstruction error in the interpreter)