diff --git a/src/qibocal/protocols/two_qubit_state_tomography.py b/src/qibocal/protocols/two_qubit_state_tomography.py
index 3b337a31a..114254edd 100644
--- a/src/qibocal/protocols/two_qubit_state_tomography.py
+++ b/src/qibocal/protocols/two_qubit_state_tomography.py
@@ -303,7 +303,7 @@ def _plot(data: StateTomographyData, fit: StateTomographyResults, target: QubitP
         return [fig_measurements], fitting_report
 
     measured = np.array(fit.measured_density_matrix_real[target]) + 1j * np.array(
-        fit.measured_raw_density_matrix_imag[target]
+        fit.measured_density_matrix_imag[target]
     )
     fig = plot_reconstruction(data.ideal[target], measured)