1 files changed, 3 insertions, 0 deletions
diff --git a/src/simulation/twoxtwo_level.rs b/src/simulation/twoxtwo_level.rs
index 8a34167..62425c8 100644
--- a/src/simulation/twoxtwo_level.rs
+++ b/src/simulation/twoxtwo_level.rs
@@ -25,11 +25,14 @@ impl State for TwoXTwoLevel {
         for val in conjugate.iter_mut() {
             *val = val.conj();
         }
+        let v = conjugate.dot(&X3().dot(&self.state))[(0, 0)];
 
         vec![
             conjugate.dot(&X0().dot(&self.state))[(0, 0)].re,
             conjugate.dot(&X1().dot(&self.state))[(0, 0)].re,
             conjugate.dot(&X2().dot(&self.state))[(0, 0)].re,
+            v.re,
+            v.im,
         ]
     }