Fixes corner_plot prior for rescaled params

pyfstat/mcmc_based_searches.py

@@ -571,7 +571,7 @@ class MCMCSearch(core.BaseSearchClass):
             fig.subplots_adjust(hspace=0.05, wspace=0.05)
 
             if add_prior:
-                self.add_prior_to_corner(axes, samples_plt)
+                self.add_prior_to_corner(axes, self.samples)
 
             fig_triangle.savefig('{}/{}_corner.png'.format(
                 self.outdir, self.label), dpi=dpi)
@@ -587,8 +587,8 @@ class MCMCSearch(core.BaseSearchClass):
             subtractor = self.get_rescale_subtractor_for_key(key)
             ax2 = ax.twinx()
             ax2.get_yaxis().set_visible(False)
-            ax2.plot(x, [(prior(xi)-subtractor)*multiplier for xi in x], '-r')
-            ax.set_xlim(xlim)
+            ax2.plot((x-subtractor)*multiplier, [prior(xi) for xi in x], '-r')
+            ax2.set_xlim(xlim)
 
     def plot_prior_posterior(self, normal_stds=2):
         """ Plot the posterior in the context of the prior """