diff --git a/CDTools/datasets/polarized_ptycho_2d_dataset.py b/CDTools/datasets/polarized_ptycho_2d_dataset.py
index 62c3a25..b1e22a7 100644
--- a/CDTools/datasets/polarized_ptycho_2d_dataset.py
+++ b/CDTools/datasets/polarized_ptycho_2d_dataset.py
@@ -69,8 +69,8 @@ class PolarizedPtycho2DDataset(Ptycho2DDataset):
         polarizer = []
         analyzer = []
         for k in range(t.tensor(translations).shape[0]):
-            polarizer.append((k//3)%3)
-            analyzer.append((k%3))
+            polarizer.append((k//3)%3 * 45)
+            analyzer.append((k%3 * 45))
         self.polarizer = t.tensor(polarizer)
         self.analyzer = t.tensor(analyzer)